diff --git a/src/BuildScriptGenerator/Node/NodePlatform.cs b/src/BuildScriptGenerator/Node/NodePlatform.cs
index ced5893d80..18bcbc1328 100644
--- a/src/BuildScriptGenerator/Node/NodePlatform.cs
+++ b/src/BuildScriptGenerator/Node/NodePlatform.cs
@@ -650,6 +650,10 @@ private static void GetAppOutputDirPath(dynamic packageJson, Dictionary<string,
             {
                 outputDirPath = "build";
             }
+            if (buildCommand.Contains("vite build", StringComparison.OrdinalIgnoreCase))
+            {
+                outputDirPath = "dist";
+            }
             else if (buildCommand.Contains("next build", StringComparison.OrdinalIgnoreCase))
             {
                 outputDirPath = ".next";
@@ -720,4 +724,4 @@ private string GetVersionUsingHierarchicalRules(string detectedVersion)
             return versionInfo.DefaultVersion;
         }
     }
-}
\ No newline at end of file
+}