diff --git a/.github/workflows/test-execution.yml b/.github/workflows/test-execution.yml
index af20594c1..76e7cf436 100644
--- a/.github/workflows/test-execution.yml
+++ b/.github/workflows/test-execution.yml
@@ -24,7 +24,7 @@ jobs:
         python -m pip install --upgrade pip
         pip install torch torchvision torchaudio --index-url https://download.pytorch.org/whl/cpu
         pip install -r requirements.txt
-        pip install websocket-client
+        pip install -r tests-unit/requirements.txt
     - name: Run Execution Tests
       run: |
-        python -m pytest tests/execution -v
\ No newline at end of file
+        python -m pytest tests/execution -v