diff --git a/projects/mmdet3d_plugin/datasets/nuscenes_vad_dataset.py b/projects/mmdet3d_plugin/datasets/nuscenes_vad_dataset.py
index 9f73a76..e2d8139 100644
--- a/projects/mmdet3d_plugin/datasets/nuscenes_vad_dataset.py
+++ b/projects/mmdet3d_plugin/datasets/nuscenes_vad_dataset.py
@@ -1374,7 +1374,7 @@ def get_data_info(self, index):
         translation = input_dict['ego2global_translation']
         can_bus = input_dict['can_bus']
         can_bus[:3] = translation
-        can_bus[3:7] = rotation
+        can_bus[3:7] = rotation.q
         patch_angle = quaternion_yaw(rotation) / np.pi * 180
         if patch_angle < 0:
             patch_angle += 360