From e4887f823a39ab0f8118aefd24ab9742cbf05320 Mon Sep 17 00:00:00 2001
From: Erik Johnston <erik@matrix.org>
Date: Wed, 22 Jan 2025 14:18:07 +0000
Subject: [PATCH] Fixup

---
 synapse/storage/databases/state/epochs.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/synapse/storage/databases/state/epochs.py b/synapse/storage/databases/state/epochs.py
index 71b8125db4..08f3f169a7 100644
--- a/synapse/storage/databases/state/epochs.py
+++ b/synapse/storage/databases/state/epochs.py
@@ -157,7 +157,7 @@ class StateEpochDataStore:
             retcols=("state_group",),
             desc="are_state_groups_pending_deletion",
         )
-        return {row["state_group"] for row in rows}
+        return {row[0] for row in rows}
 
     async def mark_state_group_as_used(self, state_group: int) -> None:
         """Mark that a given state group is used"""