diff --git a/tests/infrastructure/programs/sre/test_application_gateway.py b/tests/infrastructure/programs/sre/test_application_gateway.py
index 0f20d83b28..d250b72f40 100644
--- a/tests/infrastructure/programs/sre/test_application_gateway.py
+++ b/tests/infrastructure/programs/sre/test_application_gateway.py
@@ -408,10 +408,12 @@ def test_application_gateway_load_distribution_policies(
 
     @pulumi.runtime.test
     def test_application_gateway_location(
-        self, application_gateway_component: SREApplicationGatewayComponent
+        self,
+        application_gateway_component: SREApplicationGatewayComponent,
+        location: str,
     ):
         application_gateway_component.application_gateway.location.apply(
-            partial(assert_equal, None),
+            partial(assert_equal, location),
             run_with_unknowns=True,
         )