diff --git a/pyproject.toml b/pyproject.toml
index c2a51bad4..1740431f3 100644
--- a/pyproject.toml
+++ b/pyproject.toml
@@ -27,7 +27,8 @@ dependencies = [
     "tensorflow~=2.16.1",
     "tensorflow_datasets<=4.9.0",
     "scipy==1.10.1",
-    "websocket-client"
+    "websocket-client",
+    "websockets"
 ]
 classifiers = [
     "Development Status :: 3 - Alpha",
diff --git a/requirements.txt b/requirements.txt
index 06f95213a..5de31e12e 100644
--- a/requirements.txt
+++ b/requirements.txt
@@ -15,4 +15,5 @@ numpy~=1.26.2
 tensorflow~=2.16.1
 tensorflow_datasets<=4.9.0
 scipy==1.10.1
-websocket-client
\ No newline at end of file
+websocket-client
+websockets
\ No newline at end of file