diff --git a/ecl/codes.py b/ecl/codes.py
index 95028d851..85eb0325d 100644
--- a/ecl/codes.py
+++ b/ecl/codes.py
@@ -42,7 +42,7 @@ def extract_filename_from_code(code):
                 group1 = match.group(1)
                 filename = extract_filename_from_line(group1)
                 if "__main__" in code:
-                    filename = "main.py"
+                    filename = self.main_script
                 if filename == "":  # post-processing
                     filename = extract_filename_from_code(code)
                 assert filename != ""