diff --git a/pcdet/datasets/augmentor/data_augmentor.py b/pcdet/datasets/augmentor/data_augmentor.py
index 0945f1de6..fd0fef648 100644
--- a/pcdet/datasets/augmentor/data_augmentor.py
+++ b/pcdet/datasets/augmentor/data_augmentor.py
@@ -152,7 +152,7 @@ def random_world_translation(self, data_dict=None, config=None):
         gt_boxes[:, :3] += noise_translate
                 
         if 'roi_boxes' in data_dict.keys():
-            data_dict['roi_boxes'][:, :3] += noise_translate
+            data_dict['roi_boxes'][:, :, :3] += noise_translate
         
         data_dict['gt_boxes'] = gt_boxes
         data_dict['points'] = points