diff --git a/src/baskerville/trainer.py b/src/baskerville/trainer.py
index 91f00d9..33ec730 100644
--- a/src/baskerville/trainer.py
+++ b/src/baskerville/trainer.py
@@ -122,8 +122,8 @@ def __init__(
 
         # if log_dir is in gcs then create a local temp dir
         if is_gcs_path(self.log_dir):
-            folder_name = self.log_dir.split("/")[-1]
-            self.log_dir = tempfile.mkdtemp() + folder_name
+            folder_name = "/".join(self.log_dir.split("/")[3:])
+            self.log_dir = tempfile.mkdtemp() + "/" + folder_name
             self.gcs_log_dir = log_dir
             self.gcs = True
         else: