diff --git a/dataset/text_data.py b/dataset/text_data.py
index 01d0997..079fd0d 100644
--- a/dataset/text_data.py
+++ b/dataset/text_data.py
@@ -29,8 +29,8 @@ def __getitem__(self, idx):
         name = self.config[self.mode][idx]["name"]
         text = self.config[self.mode][idx]["text"]
 
-        # img = cv2.imread(os.path.join(self.data_path, "data", name))
-        img = cv2.imread(os.path.join(self.data_path, name))
+        img = cv2.imread(os.path.join(self.data_path, "data", name))
+        #img = cv2.imread(os.path.join(self.data_path, name))
         seq = self.text_to_seq(text)
         sample = {"img": img, "seq": seq, "seq_len": len(seq), "aug": self.mode == "train"}
         if self.transform: