diff --git a/vista2d/scripts/components.py b/vista2d/scripts/components.py
index 139247b..c1d61e2 100644
--- a/vista2d/scripts/components.py
+++ b/vista2d/scripts/components.py
@@ -90,7 +90,7 @@ def __call__(self, data):
 
             meta_data = {
                 ImageMetaKey.FILENAME_OR_OBJ: filename,
-                ImageMetaKey.SPATIAL_SHAPE: image_size,
+                ImageMetaKey.SPATIAL_SHAPE: img_array.shape[-2:],
             }
             d[key] = MetaTensor.ensure_torch_and_prune_meta(img_array, meta_data)
 
diff --git a/vista3d/cvpr_workshop/infer_cvpr.py b/vista3d/cvpr_workshop/infer_cvpr.py
index dddf7f2..3c546a6 100755
--- a/vista3d/cvpr_workshop/infer_cvpr.py
+++ b/vista3d/cvpr_workshop/infer_cvpr.py
@@ -15,7 +15,6 @@
 import os
 
 
-
 def convert_clicks(alldata):
     # indexes = list(alldata.keys())
     # data = [alldata[i] for i in indexes]