diff --git a/tools/create_data.py b/tools/create_data.py
index 05fd98cf8..f3c85fce6 100644
--- a/tools/create_data.py
+++ b/tools/create_data.py
@@ -162,8 +162,8 @@ def sunrgbd_data_prep(root_path, info_prefix, out_dir, workers):
         root_path, info_prefix, out_dir, workers=workers)
     info_train_path = osp.join(out_dir, f'{info_prefix}_infos_train.pkl')
     info_val_path = osp.join(out_dir, f'{info_prefix}_infos_val.pkl')
-    update_pkl_infos('scannet', out_dir=out_dir, pkl_path=info_train_path)
-    update_pkl_infos('scannet', out_dir=out_dir, pkl_path=info_val_path)
+    update_pkl_infos('sunrgbd', out_dir=out_dir, pkl_path=info_train_path)
+    update_pkl_infos('sunrgbd', out_dir=out_dir, pkl_path=info_val_path)
 
 
 def waymo_data_prep(root_path,