diff --git a/index.d.ts b/index.d.ts
index 5426cee..1a3f524 100644
--- a/index.d.ts
+++ b/index.d.ts
@@ -1,3 +1,5 @@
+import type WebSocket from "ws";
+
 export class CacheItem<I = unknown> {
 	constructor(rid: string, unsubscribe: (item: CacheItem) => void);
 	rid: string;
diff --git a/package.json b/package.json
index 5de11e2..00e45cd 100644
--- a/package.json
+++ b/package.json
@@ -43,6 +43,7 @@
     "@babel/preset-env": "^7.12.1",
     "@rollup/plugin-babel": "^5.2.1",
     "@rollup/plugin-node-resolve": "^8.4.0",
+    "@types/ws": "^8.5.3",
     "babel-jest": "^26.6.3",
     "coveralls": "^3.1.0",
     "cross-env": "^7.0.2",