diff --git a/api/main.py b/api/main.py
index 727705e..f2445de 100644
--- a/api/main.py
+++ b/api/main.py
@@ -1,3 +1,7 @@
+# Load environment variables
+from dotenv import load_dotenv
+load_dotenv()
+
 from contextlib import asynccontextmanager
 
 from fastapi import FastAPI
diff --git a/open_notebook/__init__.py b/open_notebook/__init__.py
index bf6bd6c..e69de29 100644
--- a/open_notebook/__init__.py
+++ b/open_notebook/__init__.py
@@ -1,3 +0,0 @@
-from dotenv import load_dotenv
-
-load_dotenv()