diff --git a/preprocessing-cuda.py b/preprocessing-cuda.py
index 08603cd..f58fb27 100644
--- a/preprocessing-cuda.py
+++ b/preprocessing-cuda.py
@@ -62,8 +62,8 @@ class DataLoader:
                 aug_image = fn.random_resized_crop(
                     images,
                     device='gpu',
-                    min_scale=0.8,
-                    max_scale=1.0
+                    size=(images.shape[1], images.shape[2]),
+                    random_area=(0.8, 1.0)
                 )
 
                 # Apply multiple random augmentations