diff --git a/env_wrapper/__init__.py b/env_wrapper/__init__.py
new file mode 100644
index 0000000..54b703e
--- /dev/null
+++ b/env_wrapper/__init__.py
@@ -0,0 +1,13 @@
+"""
+    env module
+"""
+
+
+def load_env():
+    import os
+
+    from dotenv import load_dotenv
+    if os.getenv("env") == "production":
+        load_dotenv(dotenv_path=".env")
+    else:
+        load_dotenv(dotenv_path="dev.env")
diff --git a/main.py b/main.py
index ee1ad87..8db10c8 100644
--- a/main.py
+++ b/main.py
@@ -7,7 +7,7 @@
 from auth import auth_blueprint
 from car import car_blueprint
 from database.database import db_session, init_db
-from env import load_env
+from env_wrapper import load_env
 from parking_lot import parking_lot_blueprint
 
 # load env
diff --git a/mock/run.py b/mock/run.py
index b070ab5..ed9c492 100644
--- a/mock/run.py
+++ b/mock/run.py
@@ -3,7 +3,7 @@
 """
 
 if __name__ == "__main__":
-    from env import load_env
+    from env_wrapper import load_env
     load_env()
     from mock.mock import Mock
     Mock.clean_db()