diff --git a/utils/camera_utils.py b/utils/camera_utils.py
index 1a54d0a..bdc2e9b 100644
--- a/utils/camera_utils.py
+++ b/utils/camera_utils.py
@@ -43,7 +43,7 @@ def loadCam(args, id, cam_info, resolution_scale):
     gt_image = resized_image_rgb[:3, ...]
     loaded_mask = None
 
-    if resized_image_rgb.shape[1] == 4:
+    if resized_image_rgb.shape[0] == 4:
         loaded_mask = resized_image_rgb[3:4, ...]
 
     return Camera(colmap_id=cam_info.uid, R=cam_info.R, T=cam_info.T,