diff --git a/package-lock.json b/package-lock.json
index cb2650f1..31ea3a55 100644
--- a/package-lock.json
+++ b/package-lock.json
@@ -1,12 +1,12 @@
 {
   "name": "@matrixai/ws",
-  "version": "2.0.1",
+  "version": "2.0.2",
   "lockfileVersion": 3,
   "requires": true,
   "packages": {
     "": {
       "name": "@matrixai/ws",
-      "version": "2.0.1",
+      "version": "2.0.2",
       "license": "Apache-2.0",
       "dependencies": {
         "@matrixai/async-cancellable": "^2.0.0",
diff --git a/package.json b/package.json
index 241f5df5..1a28f4ee 100644
--- a/package.json
+++ b/package.json
@@ -1,6 +1,6 @@
 {
   "name": "@matrixai/ws",
-  "version": "2.0.1",
+  "version": "2.0.2",
   "author": "Matrix AI",
   "contributors": [
     {
@@ -36,7 +36,7 @@
     "#*": "./dist/*"
   },
   "scripts": {
-    "prepare": "tsc -p ./tsconfig.build.json",
+    "prepare": "tsc -p ./tsconfig.build.json && patch-package",
     "build": "shx rm -rf ./dist && tsc -p ./tsconfig.build.json",
     "tsx": "tsx",
     "test": "node ./scripts/test.mjs",
@@ -44,8 +44,7 @@
     "lintfix": "eslint '{src,tests,scripts,benches}/**/*.{js,mjs,ts,mts,jsx,tsx}' --fix",
     "lint-shell": "find ./src ./tests ./scripts -type f -regextype posix-extended -regex '.*\\.(sh)' -exec shellcheck {} +",
     "docs": "shx rm -rf ./docs && typedoc --gitRevision master --tsconfig ./tsconfig.build.json --out ./docs src",
-    "bench": "rimraf ./benches/results && tsx ./benches/index.ts",
-    "pretest": "patch-package"
+    "bench": "rimraf ./benches/results && tsx ./benches/index.ts"
   },
   "dependencies": {
     "@matrixai/async-cancellable": "^2.0.0",