diff --git a/cmake/onnxruntime_unittests.cmake b/cmake/onnxruntime_unittests.cmake
index ac6633a3fc..2c5b61e8ea 100644
--- a/cmake/onnxruntime_unittests.cmake
+++ b/cmake/onnxruntime_unittests.cmake
@@ -185,6 +185,8 @@ if (onnxruntime_ENABLE_TRAINING)
       )
     list(APPEND onnxruntime_test_providers_src ${orttraining_test_trainingops_cuda_src})
   endif()
+endif()
+
 if (onnxruntime_USE_DNNL)
   file(GLOB_RECURSE onnxruntime_test_providers_dnnl_src CONFIGURE_DEPENDS
     "${TEST_SRC_DIR}/providers/dnnl/*"