diff --git a/IGEV-Stereo/core/utils/augmentor.py b/IGEV-Stereo/core/utils/augmentor.py
index 256f8ec..fa81ba9 100644
--- a/IGEV-Stereo/core/utils/augmentor.py
+++ b/IGEV-Stereo/core/utils/augmentor.py
@@ -302,6 +302,7 @@ class SparseFlowAugmentor:
         img2 = img2[y0:y0+self.crop_size[0], x0:x0+self.crop_size[1]]
         flow = flow[y0:y0+self.crop_size[0], x0:x0+self.crop_size[1]]
         valid = valid[y0:y0+self.crop_size[0], x0:x0+self.crop_size[1]]
+        
         return img1, img2, flow, valid