diff --git a/backend/tests/services/dataset_review/test_superset_matrix.py b/backend/tests/services/dataset_review/test_superset_matrix.py
index 3fddabe4..a948a1e1 100644
--- a/backend/tests/services/dataset_review/test_superset_matrix.py
+++ b/backend/tests/services/dataset_review/test_superset_matrix.py
@@ -9,6 +9,10 @@
 from types import SimpleNamespace
 from unittest.mock import MagicMock
 
+# Import models to ensure proper SQLAlchemy registration
+from src.models.auth import User
+from src.models.dataset_review import CompiledPreview
+
 from src.core.utils.superset_compilation_adapter import (
     PreviewCompilationPayload,
     SqlLabLaunchPayload,