diff --git a/official/cv/fastscnn/src/dataloader.py b/official/cv/fastscnn/src/dataloader.py
index b6a770a6309fd4a3878827082266e1c241a4df5d..27578e05e1c45776ad8e1bf7442b06f29d56b8fb 100644
--- a/official/cv/fastscnn/src/dataloader.py
+++ b/official/cv/fastscnn/src/dataloader.py
@@ -40,8 +40,8 @@ class CitySegmentation(SegmentationDataset):
     --------
     >>> # Transforms for Normalization
     >>> input_transform = transforms.Compose([
-    >>>     transforms.ToTensor(),
-    >>>     transforms.Normalize((.485, .456, .406), (.229, .224, .225)),
+    >>>     CV.ToTensor(),
+    >>>     CV.Normalize((.485, .456, .406), (.229, .224, .225), is_hwc=False),
     >>> ])
     >>> # Create Dataset
     >>> trainset = CitySegmentation(split='train', transform=input_transform)