diff --git a/.vsts-ci/templates/ci-general.yml b/.vsts-ci/templates/ci-general.yml
index 1419e4d3fa..a180423d6d 100644
--- a/.vsts-ci/templates/ci-general.yml
+++ b/.vsts-ci/templates/ci-general.yml
@@ -46,8 +46,10 @@ steps:
   inputs:
     targetType: inline
     script: |
+      Install-Module -Name PowerShellGet -AllowPrerelease -Scope CurrentUser -Force
+      Register-PSResourceRepository -PSGallery
       Get-Module -ListAvailable Pester
-      Install-Module InvokeBuild -Scope CurrentUser -Force
+      Install-Module -Name InvokeBuild -Scope CurrentUser -Force
       Invoke-Build
       Write-Host "##vso[task.setvariable variable=vsixPath]$(Resolve-Path powershell-*.vsix)"
     workingDirectory: $(Build.SourcesDirectory)/vscode-powershell