diff --git a/.github/workflows/unit_tests.yml b/.github/workflows/unit_tests.yml
index 07abebb1..c2c664a6 100644
--- a/.github/workflows/unit_tests.yml
+++ b/.github/workflows/unit_tests.yml
@@ -93,10 +93,8 @@ jobs:
         run: |
           uv run pytest \
             tests/test_graphiti_mock.py \
-            tests/test_graphiti_int.py \
             tests/test_node_int.py \
             tests/test_edge_int.py \
-            tests/test_entity_exclusion_int.py \
             tests/cross_encoder/test_bge_reranker_client_int.py \
             tests/driver/test_falkordb_driver.py \
             -m "not integration"
diff --git a/tests/test_entity_exclusion_int.py b/tests/test_entity_exclusion_int.py
index 53982bc9..0ac9897c 100644
--- a/tests/test_entity_exclusion_int.py
+++ b/tests/test_entity_exclusion_int.py
@@ -23,6 +23,7 @@ from graphiti_core.graphiti import Graphiti
 from graphiti_core.helpers import validate_excluded_entity_types
 from tests.helpers_test import drivers, get_driver
 
+pytestmark = pytest.mark.integration
 pytest_plugins = ('pytest_asyncio',)
 
 
diff --git a/tests/test_graphiti_int.py b/tests/test_graphiti_int.py
index 766828e3..90033f35 100644
--- a/tests/test_graphiti_int.py
+++ b/tests/test_graphiti_int.py
@@ -25,6 +25,7 @@ from graphiti_core.search.search_helpers import search_results_to_context_string
 from graphiti_core.utils.datetime_utils import utc_now
 from tests.helpers_test import GraphProvider
 
+pytestmark = pytest.mark.integration
 pytest_plugins = ('pytest_asyncio',)