diff --git a/nbs/backends/factory.ipynb b/nbs/backends/factory.ipynb
index 388bf14..e5d2302 100644
--- a/nbs/backends/factory.ipynb
+++ b/nbs/backends/factory.ipynb
@@ -26,6 +26,7 @@
    "source": [
     "#| export\n",
     "import typing as t\n",
+    "import os\n",
     "\n",
     "from ragas_experimental.backends.ragas_api_client import RagasApiClient"
    ]
diff --git a/ragas_experimental/backends/factory.py b/ragas_experimental/backends/factory.py
index f3e986c..2e8fc13 100644
--- a/ragas_experimental/backends/factory.py
+++ b/ragas_experimental/backends/factory.py
@@ -7,6 +7,7 @@
 
 # %% ../../nbs/backends/factory.ipynb 2
 import typing as t
+import os
 
 from .ragas_api_client import RagasApiClient