diff --git a/dalm/datasets/qa_gen/check_dataset.py b/dalm/datasets/qa_gen/check_dataset.py
index 8ad6309..33b687b 100644
--- a/dalm/datasets/qa_gen/check_dataset.py
+++ b/dalm/datasets/qa_gen/check_dataset.py
@@ -2,7 +2,7 @@
 
 
 # Path to the dataset saved on disk
-dataset_path = "/home/ec2-user/shamane/DALM/dalm/datasets/qa_gen/out/question_answer_pairs_train"
+dataset_path = "out/question_answer_pairs_train"
 
 # Load the dataset
 dataset = load_from_disk(dataset_path)