diff --git a/src/anemoi/inference/runner.py b/src/anemoi/inference/runner.py
index b5cba45..e7a67c2 100644
--- a/src/anemoi/inference/runner.py
+++ b/src/anemoi/inference/runner.py
@@ -296,6 +296,10 @@ def forecast(self, lead_time, input_tensor_numpy, input_state):
 
             yield result
 
+            # No need to prepare next input tensor if we are at the last step
+            if s == steps - 1:
+                continue
+
             # Update  tensor for next iteration
 
             check[:] = reset