diff --git a/.github/workflows/create-release-n-publish.yml b/.github/workflows/create-release-n-publish.yml
index bf674c6b..7b7c3629 100644
--- a/.github/workflows/create-release-n-publish.yml
+++ b/.github/workflows/create-release-n-publish.yml
@@ -102,7 +102,9 @@ jobs:
       - name: Test pip install multiple adapters
         run: pip install flowcept[mlflow,tensorboard]
       - name: Install our dependencies
-        run: pip install flowcept[all]  # This will install all dependencies, for all adapters and dev deps.
+        run: pip install flowcept[all]
+      - name: Install ml_dev dependencies
+        run: pip install flowcept[ml_dev]
       - name: Pip list
         run: pip list      
       - name: Run Docker Compose
diff --git a/.github/workflows/run-tests-kafka.yml b/.github/workflows/run-tests-kafka.yml
index 98773500..a7aedc49 100644
--- a/.github/workflows/run-tests-kafka.yml
+++ b/.github/workflows/run-tests-kafka.yml
@@ -24,6 +24,7 @@ jobs:
         run: |
           python -m pip install --upgrade pip
           python -m pip install .[all]
+          python -m pip install .[ml_dev]
 
       - name: Run docker compose
         run: docker compose -f deployment/compose-kafka.yml up -d
diff --git a/.github/workflows/run-tests-py11.yml b/.github/workflows/run-tests-py11.yml
index d1321b30..29bbefc5 100644
--- a/.github/workflows/run-tests-py11.yml
+++ b/.github/workflows/run-tests-py11.yml
@@ -56,6 +56,7 @@ jobs:
         run: |
           python -m pip install --upgrade pip
           python -m pip install .[all]
+          python -m pip install .[ml_dev]
 
       - name: List installed packages
         run: pip list