diff --git a/main.py b/main.py
index 315f6e6..18bc815 100755
--- a/main.py
+++ b/main.py
@@ -1,3 +1,4 @@
+import os
 import webui as mrq
 
 if 'TORTOISE_MODELS_DIR' not in os.environ: