diff --git a/mindspore/lite/CMakeLists.txt b/mindspore/lite/CMakeLists.txt
index f04d5515530d4ae82e7523a737bdefe9fa938b7a..37decbb3e8b7a2b361cedd9475cb79783900646f 100644
--- a/mindspore/lite/CMakeLists.txt
+++ b/mindspore/lite/CMakeLists.txt
@@ -598,7 +598,7 @@ if(MSLITE_ENABLE_TOOLS)
     endif()
 endif()
 
-if(NOT WIN32 AND MSLITE_ENABLE_TESTCASES AND NOT MSLITE_MINDDATA_IMPLEMENT STREQUAL "off")
+if(NOT WIN32 AND MSLITE_ENABLE_TESTCASES)
     add_subdirectory(${CMAKE_CURRENT_SOURCE_DIR}/test)
 endif()
 
diff --git a/mindspore/lite/test/CMakeLists.txt b/mindspore/lite/test/CMakeLists.txt
index 4dfb89af484d144da12b712aa952728ac984793f..7fcd8a2e5f693359615a7d222b2eb0fb278bc6b6 100644
--- a/mindspore/lite/test/CMakeLists.txt
+++ b/mindspore/lite/test/CMakeLists.txt
@@ -144,7 +144,11 @@ target_compile_options(gmock_tests PRIVATE -fno-rtti)
 target_link_libraries(lite-test gmock_tests)
 
 if(MSLITE_ENABLE_TRAIN)
-    target_link_libraries(lite-test mindspore-lite-train minddata-lite)
+    if(MSLITE_MINDDATA_IMPLEMENT STREQUAL "off")
+        target_link_libraries(lite-test mindspore-lite-train)
+    else()
+        target_link_libraries(lite-test mindspore-lite-train minddata-lite)
+    endif()
 endif()
 
 if(PLATFORM_ARM)