diff --git a/WizardCoder/src/process_humaneval.py b/WizardCoder/src/process_humaneval.py
index 1023a09..5a687ed 100644
--- a/WizardCoder/src/process_humaneval.py
+++ b/WizardCoder/src/process_humaneval.py
@@ -51,7 +51,7 @@
                     print("================\n")
                 # print(completion)
             if "__name__ == \"__main__\"" in completion:
-                next_line = completion.index('if __name__ == "__main__":')
+                next_line = completion.index('__name__ == "__main__"')
                 completion = completion[:next_line].strip()
                 # print(completion)
             
@@ -66,4 +66,4 @@
     
 print("save to {}".format(args.out_path))
 write_jsonl(args.out_path, output)
-print(a)
\ No newline at end of file
+print(a)