diff --git a/data/dataset.py b/data/dataset.py
index c717498..7e282d0 100644
--- a/data/dataset.py
+++ b/data/dataset.py
@@ -89,9 +89,9 @@ class TrackSynDataset(torchext.BaseDataset):
         R = []
         t = []
         for tidx in track_ind:
-            disps.append(np.load(os.path.join(sample_path, f'disp0_{tidx}.npy')), allow_pickle=True)
-            R.append(np.load(os.path.join(sample_path, f'R_{tidx}.npy')), allow_pickle=True)
-            t.append(np.load(os.path.join(sample_path, f't_{tidx}.npy')), allow_pickle=True)
+            disps.append(np.load(os.path.join(sample_path, f'disp0_{tidx}.npy'), allow_pickle=True))
+            R.append(np.load(os.path.join(sample_path, f'R_{tidx}.npy'), allow_pickle=True))
+            t.append(np.load(os.path.join(sample_path, f't_{tidx}.npy'), allow_pickle=True))
         ret[f'disp0'] = np.stack(disps, axis=0)
         ret['R'] = np.stack(R, axis=0)
         ret['t'] = np.stack(t, axis=0)