diff --git a/model_zoo/official/cv/maskrcnn/ascend310_infer/src/main.cc b/model_zoo/official/cv/maskrcnn/ascend310_infer/src/main.cc
index 5749dab750e..e6e27f112f5 100644
--- a/model_zoo/official/cv/maskrcnn/ascend310_infer/src/main.cc
+++ b/model_zoo/official/cv/maskrcnn/ascend310_infer/src/main.cc
@@ -28,9 +28,9 @@
 #include "include/api/model.h"
 #include "include/api/types.h"
 #include "include/api/serialization.h"
-#include "include/minddata/dataset/include/vision.h"
-#include "include/minddata/dataset/include/transforms.h"
-#include "include/minddata/dataset/include/execute.h"
+#include "include/dataset/vision.h"
+#include "include/dataset/transforms.h"
+#include "include/dataset/execute.h"
 #include "../inc/utils.h"
 
 using mindspore::Context;