diff --git a/setup.cfg b/setup.cfg
index 84d91657d6ff4..cc916a036176c 100644
--- a/setup.cfg
+++ b/setup.cfg
@@ -115,6 +115,7 @@ test =
     pytest-rerunfailures==12.0
     pytest-tinybird>=0.2.0
     aws-cdk-lib>=2.88.0
+    websocket-client>=1.7.0
 
 # for developing localstack
 dev =