diff --git a/data/dataset.py b/data/dataset.py
index 7e282d0..fe7d359 100644
--- a/data/dataset.py
+++ b/data/dataset.py
@@ -77,9 +77,9 @@ class TrackSynDataset(torchext.BaseDataset):
             ambs = []
             grads = []
             for tidx in track_ind:
-                imgs.append(np.load(os.path.join(sample_path, f'im{sidx}_{tidx}.npy')), allow_pickle=True)
-                ambs.append(np.load(os.path.join(sample_path, f'ambient{sidx}_{tidx}.npy')), allow_pickle=True)
-                grads.append(np.load(os.path.join(sample_path, f'grad{sidx}_{tidx}.npy')), allow_pickle=True)
+                imgs.append(np.load(os.path.join(sample_path, f'im{sidx}_{tidx}.npy'), allow_pickle=True))
+                ambs.append(np.load(os.path.join(sample_path, f'ambient{sidx}_{tidx}.npy'), allow_pickle=True))
+                grads.append(np.load(os.path.join(sample_path, f'grad{sidx}_{tidx}.npy'), allow_pickle=True))
             ret[f'im{sidx}'] = np.stack(imgs, axis=0)
             ret[f'ambient{sidx}'] = np.stack(ambs, axis=0)
             ret[f'grad{sidx}'] = np.stack(grads, axis=0)