diff --git a/data_processor/steps/image_modification/processor.py b/data_processor/steps/image_modification/processor.py
index 4322e00d..a5907db3 100644
--- a/data_processor/steps/image_modification/processor.py
+++ b/data_processor/steps/image_modification/processor.py
@@ -554,12 +554,7 @@ def mm_example_to_feature(self, example, download_fn=None):
                 raise e
             if self.variable_resolution:
                 images = np.zeros(
-                    [
-                        4,
-                        3
-                        * (self.image_preprocess.patch_size**2)
-                        * self.image_preprocess.temporal_conv_size,
-                    ],
+                    [4, 3 * (self.image_preprocess.patch_size**2)],
                     dtype=self.image_dtype,
                 )
                 grid_thw = np.array([[1, 2, 2]])