1 files changed, 23 insertions, 7 deletions
diff --git a/tests/managers/thl/test_ledger/test_lm_tx_entries.py b/tests/managers/thl/test_ledger/test_lm_tx_entries.py
index 5bf1c48..03c6e02 100644
--- a/tests/managers/thl/test_ledger/test_lm_tx_entries.py
+++ b/tests/managers/thl/test_ledger/test_lm_tx_entries.py
@@ -1,25 +1,41 @@
-from generalresearch.models.thl.ledger import LedgerEntry
+from __future__ import annotations
+
+from typing import TYPE_CHECKING
+
+from generalresearch.models.thl.ledger import (
+    LedgerEntry,
+)
+
+if TYPE_CHECKING:
+    from generalresearch.managers.thl.ledger_manager.ledger import LedgerManager
+    from generalresearch.models.thl.ledger import (
+        LedgerTransaction,
+    )
 
 
 class TestLedgerEntryManager:
 
-    def test_get_tx_entries_by_tx(self, ledger_tx, lm):
+    def test_get_tx_entries_by_tx(
+        self, ledger_tx: LedgerTransaction, ledger_manager: LedgerManager
+    ):
         # First confirm the Ledger TX exists with 2 Entries
-        res = lm.get_tx_by_id(transaction_id=ledger_tx.id)
+        res = ledger_manager.get_tx_by_id(transaction_id=ledger_tx.id)
         assert len(res.entries) == 2
 
-        tx_entries = lm.get_tx_entries_by_tx(transaction=ledger_tx)
+        tx_entries = ledger_manager.get_tx_entries_by_tx(transaction=ledger_tx)
         assert len(tx_entries) == 2
 
         assert res.entries == tx_entries
         assert isinstance(tx_entries[0], LedgerEntry)
 
-    def test_get_tx_entries_by_txs(self, ledger_tx, lm):
+    def test_get_tx_entries_by_txs(
+        self, ledger_tx: LedgerTransaction, ledger_manager: LedgerManager
+    ):
         # First confirm the Ledger TX exists with 2 Entries
-        res = lm.get_tx_by_id(transaction_id=ledger_tx.id)
+        res = ledger_manager.get_tx_by_id(transaction_id=ledger_tx.id)
         assert len(res.entries) == 2
 
-        tx_entries = lm.get_tx_entries_by_txs(transactions=[ledger_tx])
+        tx_entries = ledger_manager.get_tx_entries_by_txs(transactions=[ledger_tx])
         assert len(tx_entries) == 2
 
         assert res.entries == tx_entries