diff --git a/ch06/03_bonus_imdb-classification/train-gpt.py b/ch06/03_bonus_imdb-classification/train-gpt.py
index 1e30568..2f47ece 100644
--- a/ch06/03_bonus_imdb-classification/train-gpt.py
+++ b/ch06/03_bonus_imdb-classification/train-gpt.py
@@ -286,6 +286,8 @@ if __name__ == "__main__":
     # Instantiate dataloaders
     ###############################
 
+    base_path = Path(".")
+
     tokenizer = tiktoken.get_encoding("gpt2")
 
     train_dataset = None