diff --git a/.github/workflows/run_tests.yml b/.github/workflows/run_tests.yml
index 4c778e9b8..8530d07c2 100644
--- a/.github/workflows/run_tests.yml
+++ b/.github/workflows/run_tests.yml
@@ -109,7 +109,7 @@ jobs:
         run: |
 
           python -m pip install git+https://github.com/billschereriii/smartredis.git@multidb
-          python -m pip install .[dev,ml]
+          python -m pip install .[dev,ml-cpu]
 
 
       - name: Install ML Runtimes with Smart (with pt, tf, and onnx support)