diff --git a/src/main.py b/src/main.py
index d48da20..281310f 100644
--- a/src/main.py
+++ b/src/main.py
@@ -28,7 +28,7 @@ logger.info("Flask app initialized.")
 
 # Initialize database
 logger.info("Initializing database...")
-db = database.Database('app.db')
+db = database.Database(os.getenv('DB_LOCATION', 'app.db'))
 
 if db.get_user('SYSTEM') is None:
     logger.info("Running first time setup...")