diff --git a/main.py b/main.py
index d44299c..c9a8760 100644
--- a/main.py
+++ b/main.py
@@ -14,13 +14,14 @@ def create_app():
         "http://127.0.0.1:3000"
     ]
     CORS(app, supports_credentials=True, origins=origins)
+    
+    env = os.environ.get('FLASK_ENV', 'default')
 
     if env == "production":
         app.config.update(SESSION_COOKIE_SAMESITE="None", SESSION_COOKIE_SECURE=True)
     else:
         app.config.update(SESSION_COOKIE_SAMESITE="Lax", SESSION_COOKIE_SECURE=False)
 
-    env = os.environ.get('FLASK_ENV', 'default')
     app.config.from_object(config[env])
 
     app.register_blueprint(sse, url_prefix='/stream')