diff --git a/tests/ut/python/dataset/test_rgb_hsv.py b/tests/ut/python/dataset/test_rgb_hsv.py
index e887acb01d..14ab9ddc2d 100644
--- a/tests/ut/python/dataset/test_rgb_hsv.py
+++ b/tests/ut/python/dataset/test_rgb_hsv.py
@@ -133,6 +133,7 @@ def test_rgb_hsv_pipeline():
     # First dataset
     transforms1 = [
         vision.Decode(),
+        vision.Resize([64, 64]),
         vision.ToTensor()
     ]
     transforms1 = vision.ComposeOp(transforms1)
@@ -142,6 +143,7 @@ def test_rgb_hsv_pipeline():
     # Second dataset
     transforms2 = [
         vision.Decode(),
+        vision.Resize([64, 64]),
         vision.ToTensor(),
         vision.RgbToHsv(),
         vision.HsvToRgb()