diff --git a/cmake/winml.cmake b/cmake/winml.cmake
index 2726e7a926..5fcc6fa2f0 100644
--- a/cmake/winml.cmake
+++ b/cmake/winml.cmake
@@ -120,11 +120,30 @@ target_link_libraries(winml_lib_telemetry PRIVATE wil)
 # Add winml_adapter
 ###########################
 
-file(GLOB winml_adapter_files CONFIGURE_DEPENDS
-    "${winml_adapter_dir}/*.h"
-    "${winml_adapter_dir}/*.cpp"
+list(APPEND winml_adapter_files
+    ${winml_adapter_dir}/CpuOrtSessionBuilder.cpp
+    ${winml_adapter_dir}/CpuOrtSessionBuilder.h
+    ${winml_adapter_dir}/CustomRegistryHelper.h
+    ${winml_adapter_dir}/FeatureDescriptorFactory.cpp
+    ${winml_adapter_dir}/FeatureDescriptorFactory.h
+    ${winml_adapter_dir}/LotusEnvironment.cpp
+    ${winml_adapter_dir}/LotusEnvironment.h
+    ${winml_adapter_dir}/pch.h
+    ${winml_adapter_dir}/WinMLAdapter.cpp
+    ${winml_adapter_dir}/WinMLAdapter.h
+    ${winml_adapter_dir}/ZeroCopyInputStreamWrapper.cpp
+    ${winml_adapter_dir}/ZeroCopyInputStreamWrapper.h    
     )
 
+if (onnxruntime_USE_DML)
+  list(APPEND winml_adapter_files
+      ${winml_adapter_dir}/AbiCustomRegistryImpl.cpp
+      ${winml_adapter_dir}/AbiCustomRegistryImpl.h
+      ${winml_adapter_dir}/DmlOrtSessionBuilder.cpp
+      ${winml_adapter_dir}/DmlOrtSessionBuilder.h
+      )
+endif(onnxruntime_USE_DML)
+
 add_library(winml_adapter ${winml_adapter_files})
 
 # wil requires C++17
diff --git a/cmake/winml_unittests.cmake b/cmake/winml_unittests.cmake
index b7e409792f..18e6f7cdf8 100644
--- a/cmake/winml_unittests.cmake
+++ b/cmake/winml_unittests.cmake
@@ -44,7 +44,7 @@ function(add_winml_test)
   if (_UT_DEPENDS)
     add_dependencies(${_UT_TARGET} ${_UT_DEPENDS})
   endif()
-  target_link_libraries(${_UT_TARGET} PRIVATE ${_UT_LIBS} gtest windowsapp winml_lib_image ${onnxruntime_EXTERNAL_LIBRARIES})
+  target_link_libraries(${_UT_TARGET} PRIVATE ${_UT_LIBS} gtest windowsapp winml_lib_image ${onnxruntime_EXTERNAL_LIBRARIES} winml_lib_telemetry)
 
   add_test(NAME ${_UT_TARGET}
     COMMAND ${_UT_TARGET}