diff --git a/hub_api/config.py b/hub_api/config.py
index be8e79a..626563e 100644
--- a/hub_api/config.py
+++ b/hub_api/config.py
@@ -15,10 +15,10 @@ class Config:
   def __init__(self):
     db_name = None
     app_env = None
-    if os.getenv("FLASK_ENV") == 'production':
+    if os.getenv("FLASK_DEBUG") == 'production':
       db_name = 'hub_prod'
       app_env = 'PROD'
-    elif os.getenv("FLASK_ENV") == 'testing':
+    elif os.getenv("FLASK_DEBUG") == 'testing':
       db_name = 'test_db'
       app_env = 'TEST'
 
diff --git a/tests/base_test.py b/tests/base_test.py
index cdc25b5..627aef1 100644
--- a/tests/base_test.py
+++ b/tests/base_test.py
@@ -29,7 +29,7 @@ class BaseTest(TestCase):
 
   @classmethod
   def setUpClass(cls):
-    os.environ['FLASK_ENV'] = 'testing'
+    os.environ['FLASK_DEBUG'] = 'testing'
     cls.app = app
     cls.client = cls.app.test_client()