diff --git a/.github/workflows/ci.yml b/.github/workflows/ci.yml
index 366aaf67..39f3aea9 100644
--- a/.github/workflows/ci.yml
+++ b/.github/workflows/ci.yml
@@ -36,7 +36,7 @@ jobs:
 
     - name: Run unittest tests with coverage
       run: |
-        coverage run --source=autogpt -m unittest discover tests
+        pytest --cov=autogpt --without-integration --without-slow-integration
 
     - name: Generate coverage report
       run: |
diff --git a/tests/local_cache_test.py b/tests/local_cache_test.py
index fa596320..bb108626 100644
--- a/tests/local_cache_test.py
+++ b/tests/local_cache_test.py
@@ -4,6 +4,8 @@ import os
 import sys
 import unittest
 
+import pytest
+
 from autogpt.memory.local import LocalCache
 
 
@@ -21,6 +23,7 @@ def mock_config() -> dict:
     )
 
 
+@pytest.mark.integration_test
 class TestLocalCache(unittest.TestCase):
     """Tests for LocalCache class"""