diff --git a/beginner_source/flava_finetuning_tutorial.py b/beginner_source/flava_finetuning_tutorial.py
index 92bc5031709..ef54c5ea06d 100644
--- a/beginner_source/flava_finetuning_tutorial.py
+++ b/beginner_source/flava_finetuning_tutorial.py
@@ -172,7 +172,7 @@ def transform(tokenizer, input):
     loss.backward()
     optimizer.step()
     print(f"Loss at step {idx} = {loss}")
-    if idx > MAX_STEPS-1:
+    if idx >= MAX_STEPS-1:
       break