1 files changed, 13 insertions, 3 deletions
diff --git a/tests/managers/thl/test_buyer.py b/tests/managers/thl/test_buyer.py
index 69ea105..0ab2d52 100644
--- a/tests/managers/thl/test_buyer.py
+++ b/tests/managers/thl/test_buyer.py
@@ -1,14 +1,24 @@
-from generalresearch.models import Source
+from __future__ import annotations
+
+from collections.abc import Callable
+from typing import TYPE_CHECKING
+
+from generalresearch.models.definitions import Source
+
+if TYPE_CHECKING:
+    from generalresearch.managers.thl.buyer import BuyerManager
 
 
 class TestBuyer:
 
     def test(
         self,
-        delete_buyers_surveys,
-        buyer_manager,
+        delete_buyers_surveys: Callable[..., None],
+        buyer_manager: BuyerManager,
     ):
 
+        delete_buyers_surveys()
+
         bs = buyer_manager.bulk_get_or_create(source=Source.TESTING, codes=["a", "b"])
         assert len(bs) == 2
         buyer_a = bs[0]