diff --git a/mindformers/dataset/transforms/vision_transforms.py b/mindformers/dataset/transforms/vision_transforms.py
index d53b039f0de00740350893e8b307ab8d828f5d53..fd0a2df1aab287acd6cf3a4371b43cf901d46299 100644
--- a/mindformers/dataset/transforms/vision_transforms.py
+++ b/mindformers/dataset/transforms/vision_transforms.py
@@ -85,7 +85,7 @@ class BatchResize:
 
     def __init__(self, image_resolution, interpolation='cubic'):
         self.interpolation = INTERPOLATION.get(interpolation)
-        self.resize = vision.c_transforms.Resize(image_resolution, self.interpolation)
+        self.resize = vision.transforms.Resize(image_resolution, self.interpolation)
 
     def __call__(self, image_batch):
         """