diff --git a/workload/portal-ui/portal-ui-baseline.json b/workload/portal-ui/portal-ui-baseline.json
index fb73607ff..4f454c3bc 100644
--- a/workload/portal-ui/portal-ui-baseline.json
+++ b/workload/portal-ui/portal-ui-baseline.json
@@ -605,7 +605,11 @@
                                         "publisher": "MicrosoftWindowsDesktop",
                                         "offer": "Windows-11",
                                         "sku": "21h2-avd"
-                                    }
+                                    },
+                                    "scope": {
+                                        "location": "[steps('sessionHosts').sessionHostsRegionSection.sessionHostsRegion]",
+                                        "subscriptionId": "[steps('basics').resourceScope.subscription.subscriptionId]"
+                                    }                              
                                 },
                                 {
                                     "name": "sessionHostSizeInfobox",