diff --git a/video_prediction_tools/model_modules/video_prediction/models/weatherBench3Dcnn.py b/video_prediction_tools/model_modules/video_prediction/models/weatherBench3Dcnn.py
index 699510ab1e3d4134b2c39d4220c68f285c68c562..ceba535518b3baa1579ba4634d98ba3127753b86 100644
--- a/video_prediction_tools/model_modules/video_prediction/models/weatherBench3Dcnn.py
+++ b/video_prediction_tools/model_modules/video_prediction/models/weatherBench3Dcnn.py
@@ -82,7 +82,7 @@ class WeatherBenchModel(BaseModels):
             if i == 0:
                 x_pred = self.build_model(x[:, i, :, :, :], filters, kernels)
             else:
-                x_pred = self.build_model(_x_pred, _filters, kernels)
+                x_pred = self.build_model(x_pred, filters, kernels)
             x_hat.append(x_pred)
 
         x_hat = tf.stack(x_hat)