diff --git a/aurora/model/aurora.py b/aurora/model/aurora.py
index e211f92..f734137 100644
--- a/aurora/model/aurora.py
+++ b/aurora/model/aurora.py
@@ -198,7 +198,6 @@ def forward(self, batch: Batch) -> Batch:
         )
 
         # Remove batch and history dimension from static variables.
-        B, T = next(iter(batch.surf_vars.values()))[0]
         pred = dataclasses.replace(
             pred,
             static_vars={k: v[0, 0] for k, v in batch.static_vars.items()},