1 files changed, 7 insertions, 4 deletions
diff --git a/tests/grliq/managers/test_forensic_results.py b/tests/grliq/managers/test_forensic_results.py
index a030451..86834d0 100644
--- a/tests/grliq/managers/test_forensic_results.py
+++ b/tests/grliq/managers/test_forensic_results.py
@@ -1,18 +1,21 @@
 from __future__ import annotations
 
+from collections.abc import Callable
 from typing import TYPE_CHECKING
 
 if TYPE_CHECKING:
-    from generalresearch.grliq.managers.forensic_data import GrlIqDataManager
     from generalresearch.grliq.managers.forensic_results import (
         GrlIqCategoryResultsReader,
     )
+    from generalresearch.grliq.models.forensic_data import GrlIqData
 
 
 class TestGrlIqCategoryResultsReader:
 
     def test_filter_category_results(
-        self, grliq_dm: GrlIqDataManager, grliq_crr: GrlIqCategoryResultsReader
+        self,
+        grliq_data_factory: Callable[..., GrlIqData],
+        grliq_crr: GrlIqCategoryResultsReader,
     ):
         from generalresearch.grliq.models.forensic_result import (
             GrlIqForensicCategoryResult,
@@ -20,8 +23,8 @@ class TestGrlIqCategoryResultsReader:
         )
 
         # this is just testing that it doesn't fail
-        grliq_dm.create_dummy(is_attempt_allowed=True)
-        grliq_dm.create_dummy(is_attempt_allowed=True)
+        grliq_data_factory(is_attempt_allowed=True)
+        grliq_data_factory(is_attempt_allowed=True)
 
         res = grliq_crr.filter_category_results(limit=2, phase=Phase.OFFERWALL_ENTER)[0]
         assert res.get("category_result")