diff --git a/.github/workflows/publish.yml b/.github/workflows/publish.yml
index 31b465c..2f38acc 100644
--- a/.github/workflows/publish.yml
+++ b/.github/workflows/publish.yml
@@ -77,7 +77,7 @@ jobs:
         uses: actions/upload-artifact@v3
         with:
           name: OS specific binaries
-          path: tiktoken-node/dist
+          path: tiktoken-node/tiktoken-node.*.node
           if-no-files-found: error
 
   publish:
@@ -107,7 +107,10 @@ jobs:
         uses: actions/download-artifact@v3
         with:
           name: OS specific binaries
-          path: tiktoken-node/dist/
+          path: dist/
+
+      - name: Copy .node file
+        run: cp dist/tiktoken-node.*.node tiktoken-node
 
       - name: Publish to npm
         uses: JS-DevTools/npm-publish@v1
diff --git a/tiktoken-node/package.json b/tiktoken-node/package.json
index 72810d5..03180e9 100644
--- a/tiktoken-node/package.json
+++ b/tiktoken-node/package.json
@@ -37,7 +37,7 @@
   },
   "wireit": {
     "build": {
-      "command": "napi build --platform --release ./dist",
+      "command": "napi build --platform --release",
       "files": [
         "src/**",
         "Cargo.toml",
@@ -45,8 +45,10 @@
         "build.rs"
       ],
       "output": [
-        "dist/tiktoken-node.*.node"
+        "tiktoken-node.*.node",
+        "index.js",
+        "index.d.ts"
       ]
     }
   }
-}
+}
\ No newline at end of file