diff --git a/tests/e2e/perf/test_database_upgrade.py b/tests/e2e/perf/test_database_upgrade.py
index f165b9b738f..8079aa0274d 100644
--- a/tests/e2e/perf/test_database_upgrade.py
+++ b/tests/e2e/perf/test_database_upgrade.py
@@ -55,9 +55,9 @@ def test_database_upgrade(proxy: ProxyClient, idp: Keycloak, resources: Resource
         monitor.assert_complete(state)
         assert_finished(connection, state)
         rss_after: Final = read_rss_everywhere(proxy)
-        assert {r.worker for r in rss_before.readings} == {r.worker for r in rss_after.readings}, (
-            "Gateway workers restarted"
-        )
+        assert {(r.replica, r.hostname) for r in rss_before.readings} == {
+            (r.replica, r.hostname) for r in rss_after.readings
+        }, "Gateway pods replaced during the upgrade"
         observed: Final = Observation(
             records=outcome.records,
             mock_requests=tuple(seen for url in urls for seen in read_mock_requests(url)),