diff --git a/sema4ai/tests/sema4ai_code_tests/test_compute_data_sources_state.py b/sema4ai/tests/sema4ai_code_tests/test_compute_data_sources_state.py
index 10756c4d..016cd9fa 100644
--- a/sema4ai/tests/sema4ai_code_tests/test_compute_data_sources_state.py
+++ b/sema4ai/tests/sema4ai_code_tests/test_compute_data_sources_state.py
@@ -262,7 +262,6 @@ def setup_data_source(datasource: DatasourceInfoTypedDict):
     data_regression.check(fixed_result, basename="missing_data_source_none")
 
 
-@pytest.mark.data_server
 def setup_sqlite_data_source(data_server_cli, tmpdir):
     import json
 
@@ -275,6 +274,7 @@ def setup_sqlite_data_source(data_server_cli, tmpdir):
     )
 
 
+@pytest.mark.data_server
 def test_compute_data_sources_state(
     data_server_cli: DataServerCliWrapper,
     ws_root_path,