diff --git a/backend/lambda/gb_snapshot/gb_snapshot.py b/backend/lambda/gb_snapshot/gb_snapshot.py
index 1ed4172..739e6b0 100644
--- a/backend/lambda/gb_snapshot/gb_snapshot.py
+++ b/backend/lambda/gb_snapshot/gb_snapshot.py
@@ -5,10 +5,10 @@
 from .ptypes import (
     BalancingTotals,
     EmbeddedSnapshot,
+    ForeignMarketSnapshot,
     NationalGridGspSnapshot,
     SummaryOutput,
     TotalsSnapshot,
-    UnitGroupSnapshot,
 )
 from .generation import GenerationTotals
 from .foreign_markets import ForeignMarketTotals
@@ -17,7 +17,7 @@
 from .bm import Bm
 
 
-def combine_totals(gen_totals: List[UnitGroupSnapshot], fm_totals: TotalsSnapshot):
+def combine_totals(gen_totals: List[TotalsSnapshot], fm_totals: ForeignMarketSnapshot):
     totals = gen_totals + [fm_totals]
     totals.sort(key=lambda x: x.ac, reverse=True)
     return totals