diff --git a/images/win/post-generation/Choco.ps1 b/images/win/post-generation/Choco.ps1
new file mode 100644
index 00000000..3f8e7c8d
--- /dev/null
+++ b/images/win/post-generation/Choco.ps1
@@ -0,0 +1,2 @@
+# Step to avoid initial delay for choco scripts
+choco upgrade chocolatey
\ No newline at end of file
diff --git a/images/win/post-generation/VSConfiguration.ps1 b/images/win/post-generation/VSConfiguration.ps1
index 3561e6d1..0ff9d7bf 100644
--- a/images/win/post-generation/VSConfiguration.ps1
+++ b/images/win/post-generation/VSConfiguration.ps1
@@ -1,7 +1,4 @@
 $vsInstallRoot = Get-VisualStudioPath
 $devEnvPath = "$vsInstallRoot\Common7\IDE\devenv.exe"
 
-cmd.exe /c "`"$devEnvPath`" /updateconfiguration"
-
-# Step to avoid initial delay for choco scripts
-choco upgrade chocolatey
\ No newline at end of file
+cmd.exe /c "`"$devEnvPath`" /updateconfiguration"
\ No newline at end of file