Fixed mocks broken by change in get-azureenvironment