diff --git a/src/super_gradients/training/transforms/transforms.py b/src/super_gradients/training/transforms/transforms.py
index c024b0a6d2..ba973f4153 100644
--- a/src/super_gradients/training/transforms/transforms.py
+++ b/src/super_gradients/training/transforms/transforms.py
@@ -850,7 +850,7 @@ def __call__(self, sample: dict) -> dict:
         return sample
 
     def get_equivalent_preprocessing(self) -> List[Dict]:
-        return [{Processings.DetectionRescale: {"output_shape": self.output_size}}]
+        return [{Processings.DetectionRescale: {"output_shape": self.output_shape}}]
 
 
 @register_transform(Transforms.DetectionRandomRotate90)