diff --git a/governance/pyth_staking_sdk/package.json b/governance/pyth_staking_sdk/package.json
index edbd5d55d7..d023d26449 100644
--- a/governance/pyth_staking_sdk/package.json
+++ b/governance/pyth_staking_sdk/package.json
@@ -1,14 +1,17 @@
 {
   "name": "@pythnetwork/staking-sdk",
-  "version": "0.0.1",
+  "version": "0.0.2",
   "description": "Pyth staking SDK",
   "main": "src/index.ts",
   "types": "src/index.d.ts",
+  "files": [
+    "dist/**/*"
+  ],
   "publishConfig": {
     "access": "public"
   },
   "scripts": {
-    "build": "tsc && node scripts/update-package-json.mjs",
+    "build": "tsc",
     "test": "pnpm run test:format && pnpm run test:lint && pnpm run test:integration && pnpm run test:types",
     "fix": "pnpm fix:lint && pnpm fix:format",
     "fix:format": "prettier --write .",
@@ -16,7 +19,8 @@
     "test:format": "prettier --check .",
     "test:lint": "jest --selectProjects lint",
     "test:types": "tsc",
-    "test:integration": "echo disabled # jest --selectProjects integration"
+    "test:integration": "echo disabled # jest --selectProjects integration",
+    "prepublishOnly": "node scripts/update-package-json.mjs"
   },
   "devDependencies": {
     "@cprussin/eslint-config": "^3.0.0",
diff --git a/governance/pyth_staking_sdk/scripts/update-package-json.mjs b/governance/pyth_staking_sdk/scripts/update-package-json.mjs
index 931eb14cfc..6b9fd53c02 100644
--- a/governance/pyth_staking_sdk/scripts/update-package-json.mjs
+++ b/governance/pyth_staking_sdk/scripts/update-package-json.mjs
@@ -11,10 +11,11 @@ import { fileURLToPath } from "url";
 
 const __dirname = path.dirname(fileURLToPath(import.meta.url));
 
-const distPackageJsonPath = path.join(__dirname, "..", "dist", "package.json");
+const distPackageJsonPath = path.join(__dirname, "..", "package.json");
 
 const packageJson = JSON.parse(fs.readFileSync(distPackageJsonPath, "utf8"));
 
-packageJson.main = "src/index.js";
+packageJson.main = "dist/src/index.js";
+packageJson.types = "dist/src/index.d.ts";
 
 fs.writeFileSync(distPackageJsonPath, JSON.stringify(packageJson, null, 2));