diff --git a/tests/providers/test_fidelity_planviewer.py b/tests/providers/test_fidelity_planviewer.py
index fe4feca..55b069e 100644
--- a/tests/providers/test_fidelity_planviewer.py
+++ b/tests/providers/test_fidelity_planviewer.py
@@ -105,7 +105,7 @@ def test_gains_offset_emits_deposit_when_pot_exceeds_contributions() -> None:
     offset = _gains_offset_activity(holdings, txs, as_of)
     assert offset is not None
     assert offset.activity_type in (ActivityType.DEPOSIT, ActivityType.WITHDRAWAL)
-    assert offset.amount > 0
+    assert offset.amount is not None and offset.amount > 0
     assert offset.external_id == "fidelity:gains:2026-04-18"
 
 
diff --git a/tests/providers/test_imap.py b/tests/providers/test_imap.py
index 5e1c14f..63638cb 100644
--- a/tests/providers/test_imap.py
+++ b/tests/providers/test_imap.py
@@ -46,7 +46,8 @@ def test_single_tax_year_under_cap_stays_isa() -> None:
 def test_overflow_past_cap_flips_to_gia() -> None:
     acts = [
         _buy(datetime(2024, 5, 1, tzinfo=UTC), "100", "80"),   # £8,000
-        _buy(datetime(2024, 6, 1, tzinfo=UTC), "150", "80"),   # +£12,000 → £20,000 total; prev £8k < cap → ISA
+        # +£12,000 → £20,000 total; prev £8k < cap → ISA
+        _buy(datetime(2024, 6, 1, tzinfo=UTC), "150", "80"),
         _buy(datetime(2024, 7, 1, tzinfo=UTC), "10", "80"),    # prev £20,000 ≥ cap → GIA
         _buy(datetime(2024, 8, 1, tzinfo=UTC), "10", "80"),    # GIA
     ]