diff --git a/.gitlab-ci/lava/lava_job_submitter.py b/.gitlab-ci/lava/lava_job_submitter.py
index ee36f47b401..2a262313bb2 100755
--- a/.gitlab-ci/lava/lava_job_submitter.py
+++ b/.gitlab-ci/lava/lava_job_submitter.py
@@ -609,7 +609,7 @@ class StructuredLoggerWrapper:
         return context
 
 
-if __name__ == "__main__":
+def main():
     # given that we proxy from DUT -> LAVA dispatcher -> LAVA primary -> us ->
     # GitLab runner -> GitLab primary -> user, safe to say we don't need any
     # more buffering
@@ -617,3 +617,6 @@ if __name__ == "__main__":
     sys.stderr.reconfigure(line_buffering=True)
 
     fire.Fire(LAVAJobSubmitter)
+
+if __name__ == "__main__":
+    main()