diff --git a/src/datasets/human36m.py b/src/datasets/human36m.py
index b401e78..25f406a 100644
--- a/src/datasets/human36m.py
+++ b/src/datasets/human36m.py
@@ -42,9 +42,9 @@ def __init__(self, actions, data_path, use_hg=True, is_train=True):
             self.train_3d = torch.load(os.path.join(data_path, 'train_3d.pth.tar'))
             self.train_2d = torch.load(os.path.join(data_path, train_2d_file))
             for k2d in self.train_2d.keys():
-                (sub, act, fname, cls) = k2d
+                (sub, act, fname) = k2d
                 k3d = k2d
-                k3d = (sub, act, fname[:-3], cls) if fname.endswith('-sh') else k3d
+                k3d = (sub, act, fname[:-3]) if fname.endswith('-sh') else k3d
                 num_f, _ = self.train_2d[k2d].shape
                 assert self.train_3d[k3d].shape[0] == self.train_2d[k2d].shape[0], '(training) 3d & 2d shape not matched'
                 for i in range(num_f):
@@ -56,11 +56,11 @@ def __init__(self, actions, data_path, use_hg=True, is_train=True):
             self.test_3d = torch.load(os.path.join(data_path, 'test_3d.pth.tar'))
             self.test_2d = torch.load(os.path.join(data_path, test_2d_file))
             for k2d in self.test_2d.keys():
-                (sub, act, fname, cls) = k2d
+                (sub, act, fname) = k2d
                 if act not in self.actions:
                     continue
                 k3d = k2d
-                k3d = (sub, act, fname[:-3], cls) if fname.endswith('-sh') else k3d
+                k3d = (sub, act, fname[:-3]) if fname.endswith('-sh') else k3d
                 num_f, _ = self.test_2d[k2d].shape
                 assert self.test_2d[k2d].shape[0] == self.test_3d[k3d].shape[0], '(test) 3d & 2d shape not matched'
                 for i in range(num_f):