diff --git a/data/dataset.py b/data/dataset.py
index 0a2db61..188f1b2 100644
--- a/data/dataset.py
+++ b/data/dataset.py
@@ -159,7 +159,7 @@ class Dataset(Dataset):
         if 'frame_nr' in self.signature:
             item['frame_nr'] = frame.frame_nr
         if 'pose_shape' in self.signature:
-            pose_shape = np.load(os.path.join(self.folder, 'pose_shape', frame.id + separator + frame.frame_nr + '.npy'))
+            pose_shape = np.load(os.path.join(self.folder, 'pose_shape', frame.id + separator + frame.frame_nr + '.npy')).astype(np.float32)
             pose_shape[:3] = 0
             item['pose_shape'] = pose_shape
         if 'mesh_body_posed' in self.signature: