import unit

tests/test_mint.py

@@ -2,7 +2,7 @@
 
 import pytest
 
-from cashu.core.base import BlindedMessage, MintKeyset, Proof
+from cashu.core.base import BlindedMessage, MintKeyset, Proof, Unit
 from cashu.core.crypto.b_dhke import step1_alice
 from cashu.core.helpers import calculate_number_of_blank_outputs
 from cashu.core.models import PostMintQuoteRequest
@@ -218,6 +218,7 @@ async def test_generate_change_promises_returns_empty_if_no_outputs(ledger: Ledg
 
 @pytest.mark.asyncio
 async def test_get_balance(ledger: Ledger):
+    unit = Unit["sat"]
     active_keyset: MintKeyset = next(
         filter(lambda k: k.active and k.unit == unit, ledger.keysets.values())
     )