diff --git a/.github/workflows/build.yml b/.github/workflows/build.yml
index 47a37649..7ab473b9 100644
--- a/.github/workflows/build.yml
+++ b/.github/workflows/build.yml
@@ -53,7 +53,6 @@ jobs:
       - name: scripts
         run: |
           gulp tslint
-          gulp genAikey
           vsce package
       - name: run tests
         if: ${{ matrix.os != 'windows-latest' }}
@@ -93,7 +92,6 @@ jobs:
     - name: scripts
       run: |
         gulp tslint
-        gulp genAikey
         vsce package
     - name: upload .vsix to github tag
       uses: svenstaro/upload-release-action@v2
@@ -111,4 +109,8 @@ jobs:
         fi
     - name: publish
       if: steps.check-version.outputs.ISPRODUCTION == 'true'
-      run: vsce publish -p ${{ secrets.VSCE_TOKEN }}
+      env:
+        PROD_AIKEY: ${{ secrets.PROD_AIKEY }}
+      run: | 
+        gulp genAikey
+        vsce publish -p ${{ secrets.VSCE_TOKEN }}
diff --git a/gulpfile.js b/gulpfile.js
index 7888a888..a8955511 100644
--- a/gulpfile.js
+++ b/gulpfile.js
@@ -81,20 +81,10 @@ gulp.task("clean", (done) => {
 });
 
 gulp.task("genAikey", (done) => {
-    if (process.env.TRAVIS_TAG) {
-        const ISPROD = /^v?[0-9]+\.[0-9]+\.[0-9]+$/.test(process.env.TRAVIS_TAG || "");
-        const packageJson = JSON.parse(fs.readFileSync("package.json"));
-        if (ISPROD) {
-            packageJson.aiKey = process.env["PROD_AIKEY"];
-        } else {
-            packageJson.aiKey = process.env["INT_AIKEY"] || packageJson.aiKey;
-        }
-        fs.writeFileSync("package.json", JSON.stringify(packageJson, null, 2) + "\n");
-        done();
-    } else {
-        log("Skipping genAiKey");
-        done();
-    }
+    const packageJson = JSON.parse(fs.readFileSync("package.json"));
+    packageJson.aiKey = process.env.PROD_AIKEY;
+    fs.writeFileSync("package.json", JSON.stringify(packageJson, null, 2) + "\n");
+    done();
 });
 
 gulp.task("test", (done) => {