diff --git a/images/win/scripts/ImageHelpers/VisualStudioHelpers.ps1 b/images/win/scripts/ImageHelpers/VisualStudioHelpers.ps1
index bf87912de..10e8d0cd8 100644
--- a/images/win/scripts/ImageHelpers/VisualStudioHelpers.ps1
+++ b/images/win/scripts/ImageHelpers/VisualStudioHelpers.ps1
@@ -69,6 +69,7 @@ function Get-VisualStudioInstallation {
     Param
     (
         [Parameter(Mandatory)]
+        [ValidateSet('VS','BuildTools')]
         [String] $VSInstallType
     )
 
@@ -92,6 +93,7 @@ function Get-VisualStudioComponents {
     Param
     (
         [Parameter(Mandatory)]
+        [ValidateSet('VS','BuildTools')]
         [String] $VSInstallType
     )
     (Get-VisualStudioInstallation -VSInstallType $VSInstallType).Packages | Where-Object type -in 'Component', 'Workload' |