diff --git a/gallery/plot_optical_flow.py b/gallery/plot_optical_flow.py
index 9e8d0006f1a..b0a93209877 100644
--- a/gallery/plot_optical_flow.py
+++ b/gallery/plot_optical_flow.py
@@ -180,7 +180,7 @@ def preprocess(img1_batch, img2_batch):
 #     # Note: it would be faster to predict batches of flows instead of individual flows
 #     img1, img2 = preprocess(img1, img2)
 
-#     list_of_flows = model(img1.to(device), img1.to(device))
+#     list_of_flows = model(img1.to(device), img2.to(device))
 #     predicted_flow = list_of_flows[-1][0]
 #     flow_img = flow_to_image(predicted_flow).to("cpu")
 #     output_folder = "/tmp/"  # Update this to the folder of your choice