diff --git a/server/text_generation_server/utils/gcs.py b/server/text_generation_server/utils/gcs.py
index 46099971..0f55244c 100644
--- a/server/text_generation_server/utils/gcs.py
+++ b/server/text_generation_server/utils/gcs.py
@@ -1,3 +1,4 @@
+import os
 from google.cloud import storage
 
 GCS_PREFIX = "gs://"