diff --git a/oxe_utils/transforms.py b/oxe_utils/transforms.py
index 2d97c04..a66d1ce 100644
--- a/oxe_utils/transforms.py
+++ b/oxe_utils/transforms.py
@@ -621,6 +621,7 @@ def berkeley_mvp_dataset_transform(trajectory: Dict[str, Any]) -> Dict[str, Any]
 
 
 def berkeley_rpt_dataset_transform(trajectory: Dict[str, Any]) -> Dict[str, Any]:
+    trajectory["observation"]["gripper"] = trajectory["observation"]["gripper"][:, None]
     return trajectory