diff --git a/model_zoo/official/cv/yolov3_darknet53/ascend310_infer/src/main.cc b/model_zoo/official/cv/yolov3_darknet53/ascend310_infer/src/main.cc
index bbe1a8ac38e..f737745cef7 100644
--- a/model_zoo/official/cv/yolov3_darknet53/ascend310_infer/src/main.cc
+++ b/model_zoo/official/cv/yolov3_darknet53/ascend310_infer/src/main.cc
@@ -28,9 +28,9 @@
 #include "include/api/context.h"
 #include "include/api/types.h"
 #include "include/api/serialization.h"
-#include "include/minddata/dataset/include/vision_ascend.h"
-#include "include/minddata/dataset/include/execute.h"
-#include "include/minddata/dataset/include/vision.h"
+#include "include/dataset/vision_ascend.h"
+#include "include/dataset/execute.h"
+#include "include/dataset/vision.h"
 #include "inc/utils.h"
 
 using mindspore::Context;