diff --git a/backend/tests/unit/test_modules/test_atheris_fuzzer.py b/backend/tests/unit/test_modules/test_atheris_fuzzer.py
index 0efcc76..8873aa6 100644
--- a/backend/tests/unit/test_modules/test_atheris_fuzzer.py
+++ b/backend/tests/unit/test_modules/test_atheris_fuzzer.py
@@ -159,7 +159,7 @@ class TestAtherisFuzzerFindingGeneration:
 
     async def test_create_crash_finding(self, atheris_fuzzer):
         """Test crash finding creation"""
-        from models.finding_schema import FoundBy
+        from modules.base import FoundBy
 
         finding = atheris_fuzzer.create_finding(
             rule_id="atheris-crash",
diff --git a/backend/tests/unit/test_modules/test_cargo_fuzzer.py b/backend/tests/unit/test_modules/test_cargo_fuzzer.py
index 5b79ad7..ef20b77 100644
--- a/backend/tests/unit/test_modules/test_cargo_fuzzer.py
+++ b/backend/tests/unit/test_modules/test_cargo_fuzzer.py
@@ -161,7 +161,7 @@ class TestCargoFuzzerFindingGeneration:
 
     async def test_create_finding_from_crash(self, cargo_fuzzer):
         """Test finding creation"""
-        from models.finding_schema import FoundBy
+        from modules.base import FoundBy
 
         finding = cargo_fuzzer.create_finding(
             rule_id="cargo-fuzz-crash",