diff --git a/examples/models/mobilenet_v2/model.py b/examples/models/mobilenet_v2/model.py
index 5c2c7ff7016..32e82197e46 100644
--- a/examples/models/mobilenet_v2/model.py
+++ b/examples/models/mobilenet_v2/model.py
@@ -57,7 +57,7 @@ def get_example_inputs(self):
             input_tensor = preprocess(input_image)
             input_batch = input_tensor.unsqueeze(0)
             input_batch = (input_batch,)
-        return (torch.randn(tensor_size),)
+        return input_batch
 
 
 class MV2UntrainedModel(EagerModelBase):