diff --git a/projects/visual_bert/configs/masked_coco/pretrain_train_val.yaml b/projects/visual_bert/configs/masked_coco/pretrain_train_val.yaml
index e9a4c16ba..904b16a9d 100644
--- a/projects/visual_bert/configs/masked_coco/pretrain_train_val.yaml
+++ b/projects/visual_bert/configs/masked_coco/pretrain_train_val.yaml
@@ -8,8 +8,8 @@ dataset_config:
     use_features: true
     features:
       train:
-      - coco/defaults/features/coco_trainval2014.lmdb
-      - coco/defaults/features/coco_trainval2014.lmdb
+      - coco/defaults/features/trainval2014.lmdb
+      - coco/defaults/features/trainval2014.lmdb
     annotations:
       train:
       - coco/defaults/annotations/imdb_karpathy_train_by_image.npy