diff --git a/isaacgymenvs/tasks/franka_cube_stack.py b/isaacgymenvs/tasks/franka_cube_stack.py
index 08d3915a5..470ad8b13 100644
--- a/isaacgymenvs/tasks/franka_cube_stack.py
+++ b/isaacgymenvs/tasks/franka_cube_stack.py
@@ -543,7 +543,7 @@ def _reset_init_cube_state(self, cube, env_ids, check_valid=True):
         elif cube.lower() == 'b':
             this_cube_state_all = self._init_cubeB_state
             other_cube_state = self._init_cubeA_state[env_ids, :]
-            cube_heights = self.states["cubeA_size"]
+            cube_heights = self.states["cubeB_size"]
         else:
             raise ValueError(f"Invalid cube specified, options are 'A' and 'B'; got: {cube}")