diff --git a/scripts/test_my_prompt_custom_script.py b/scripts/test_my_prompt_custom_script.py
index 115d3de..9558e19 100644
--- a/scripts/test_my_prompt_custom_script.py
+++ b/scripts/test_my_prompt_custom_script.py
@@ -73,7 +73,7 @@ class Script(scripts.Script):
                 continue
                 
             # Check for stopping_generation
-            if state.stopping_generation: # Unsure if this works, 
+            if state.stopping_generation:  
                 print("Stopping generation as requested.")
                 break