diff --git a/06_gpu_and_ml/text_generation_inference.py b/06_gpu_and_ml/text_generation_inference.py
index 66312d4de..8cc4b6eb9 100644
--- a/06_gpu_and_ml/text_generation_inference.py
+++ b/06_gpu_and_ml/text_generation_inference.py
@@ -58,7 +58,8 @@ def download_model():
             MODEL_ID,
             "--revision",
             REVISION,
-        ]
+        ],
+        check=True,
     )