diff --git a/smpl/smpl_torch.py b/smpl/smpl_torch.py
index f2a87cb..16e1855 100644
--- a/smpl/smpl_torch.py
+++ b/smpl/smpl_torch.py
@@ -218,7 +218,7 @@ class SMPL(Module):
     result = result - root_joint
     joints = joints - root_joint
 
-    if remove:
+    if remove and verts is None:
       result = result[:,self.to_keep,:]
 
     return result, joints