diff --git a/requirements.txt b/requirements.txt
index e67c903..4c5d9a9 100644
--- a/requirements.txt
+++ b/requirements.txt
@@ -5,3 +5,4 @@ docker
 javalang
 bs4
 sacrebleu
+python-dotenv
diff --git a/src/server.py b/src/server.py
index fc0956e..a040a25 100644
--- a/src/server.py
+++ b/src/server.py
@@ -8,6 +8,10 @@ from routes.datasets import router as datasets_router
 from werkzeug.exceptions import HTTPException
 import os
 
+from dotenv import load_dotenv
+
+load_dotenv()
+
 app = Flask(__name__, static_folder='../public', static_url_path='/')
 CORS(app)
 
diff --git a/src/utils/build_handlers.py b/src/utils/build_handlers.py
index 48ca468..e8eca64 100644
--- a/src/utils/build_handlers.py
+++ b/src/utils/build_handlers.py
@@ -633,7 +633,8 @@ def get_build_handler(root: str, repo: str, verbose: bool = False) -> BuildHandl
         an instance of GradleHandler or MavenHandler
     """
     path = os.path.join(root, repo)
-    return WaitBuildHander("NO REPO PATH", "NO BUILD FILE", {})
+    if os.getenv('MOCK_BUILD_HANDLER', False):
+        return WaitBuildHander("NO REPO PATH", "NO BUILD FILE", {})
 
     # 1) If it's a tarball, extract it
     if os.path.isfile(path) and tarfile.is_tarfile(path):