diff --git a/.github/workflows/run-tests.yml b/.github/workflows/run-tests.yml
index 60b2f45e..82cc8d63 100644
--- a/.github/workflows/run-tests.yml
+++ b/.github/workflows/run-tests.yml
@@ -45,10 +45,10 @@ jobs:
         export MQ_TYPE=kafka
         export MQ_PORT=9092
         python -c 'from flowcept.configs import MQ_TYPE, MQ_PORT; print(f"MQ_TYPE={MQ_TYPE}"); print(f"MQ_TYPE={MQ_PORT}")'
-        python -c 'from flowcept import FlowceptConsumerAPI; assert FlowceptConsumerAPI.services_alive()'
+        python -c 'from flowcept import Flowcept; assert Flowcept.services_alive()'
     - name: Run Tests with Kafka
       run: |
         export MQ_TYPE=kafka
         export MQ_PORT=9092
         # Ignoring heavy tests. They are executed with Kafka in another GH Action. 
-        pytest --ignore=tests/decorator_tests/ml_tests --ignore=tests/adapters/test_tensorboard.py
\ No newline at end of file
+        pytest --ignore=tests/decorator_tests/ml_tests --ignore=tests/adapters/test_tensorboard.py