diff --git a/mock/mock.ts b/mock/mock.ts
index 3c8cfc9..26aeb87 100644
--- a/mock/mock.ts
+++ b/mock/mock.ts
@@ -8,7 +8,7 @@ import { createExecutionContext, getExecutionContext } from "../src/context";
 import { startRecordingMocks } from "./utils";
 
 const PORT = 6789;
-const PROTO_FILE = "../proto/services.proto";
+const PROTO_FILE = "../../proto/services.proto";
 const packageDef = protoLoader.loadSync(path.resolve(__dirname, PROTO_FILE));
 const grpcObj = grpc.loadPackageDefinition(
   packageDef
diff --git a/package.json b/package.json
index 8ea57c5..c6945ba 100644
--- a/package.json
+++ b/package.json
@@ -3,13 +3,14 @@
   "version": "1.0.0",
   "main": "dist/src/index.js",
   "files": [
-    "dist"
+    "dist",
+    "proto/services.proto"
   ],
   "scripts": {
-    "build": "tsc && cp ./proto/services.proto ./dist/proto/",
+    "build": "tsc",
     "prepare": "npm run build",
     "commit": "cz"
-  },
+    },
   "repository": "git@github.com:keploy/typescript-sdk.git",
   "author": "Rajat Sharma <lunasunkaiser@gmail.com>",
   "license": "MIT",
diff --git a/src/keploy.ts b/src/keploy.ts
index 1de2cca..b599054 100644
--- a/src/keploy.ts
+++ b/src/keploy.ts
@@ -15,7 +15,7 @@ import { StrArr } from "../proto/services/StrArr";
 import assert = require("assert");
 import { createExecutionContext, getExecutionContext } from "./context";
 
-const PROTO_PATH = "../proto/services.proto";
+const PROTO_PATH = "../../proto/services.proto";
 const packageDef = protoLoader.loadSync(path.resolve(__dirname, PROTO_PATH));
 const grpcObj = grpc.loadPackageDefinition(
   packageDef