diff --git a/tools/nuget/generate_nuspec_for_native_nuget.py b/tools/nuget/generate_nuspec_for_native_nuget.py
index ae323fa482..dc52591343 100644
--- a/tools/nuget/generate_nuspec_for_native_nuget.py
+++ b/tools/nuget/generate_nuspec_for_native_nuget.py
@@ -167,6 +167,7 @@ def generate_files(list, args):
                           '" target="build\\native\\include" />')
 
     if includes_winml:
+        mlop_path = 'onnxruntime\\core\\providers\\dml\\dmlexecutionprovider\\inc\\mloperatorauthor.h'
         # Add microsoft.ai.machinelearning headers
         files_list.append('<file src=' + '"' + os.path.join(args.ort_build_path, args.build_config,
                                                             'microsoft.ai.machinelearning.h') +
@@ -183,8 +184,7 @@ def generate_files(list, args):
                           '" target="lib\\uap10.0\\Microsoft.AI.MachineLearning.winmd" />')
         # Add custom operator headers
         files_list.append('<file src=' + '"' +
-                          os.path.join(args.sources_path,
-                                       'onnxruntime\\core\\providers\\dml\\dmlexecutionprovider\\inc\\mloperatorauthor.h') +
+                          os.path.join(args.sources_path, mlop_path) +
                           '" target="build\\native\\include" />')
 
     # Process runtimes