diff --git a/src/log.py b/src/log.py
index e513c2e..4a41adc 100644
--- a/src/log.py
+++ b/src/log.py
@@ -1,7 +1,5 @@
 # SPDX-License-Identifier: LGPL-2.1-only
-# Copyright 2021 Jookia <contact@jookia.org>
-
-import os
+# Copyright 2022 Jookia <contact@jookia.org>
 
 LEXER = 0
 PARSER = 1
@@ -36,8 +34,7 @@
     print("%s %s: %s" % (level_name, module_name, text))
 
 
-def set_default_log_level():
-    level = os.getenv("NEWLANG_LOG_LEVEL")
+def set_default_log_level(level):
     global log_level
     if not level or level == "NORMAL":
         log_level = 0
diff --git a/src/main.py b/src/main.py
index bd8ec9a..1928875 100644
--- a/src/main.py
+++ b/src/main.py
@@ -1,6 +1,7 @@
 # SPDX-License-Identifier: LGPL-2.1-only
 # Copyright 2022 Jookia <contact@jookia.org>
 
+import os
 import sys
 
 from src import port
@@ -11,10 +12,11 @@
 
 def main():
     args = sys.argv
+    log_level = os.getenv("NEWLANG_LOG_LEVEL")
     if len(args) != 2:
         print("Usage: main.py FILENAME")
         return 1
-    log.set_default_log_level()
+    log.set_default_log_level(log_level)
     ast = oldparse.parse_file(args[1])
     if not ast:
         return 1