1 files changed, 3 insertions, 3 deletions
diff --git a/synapse/storage/state.py b/synapse/storage/state.py
index 19b16ed404..a438530071 100644
--- a/synapse/storage/state.py
+++ b/synapse/storage/state.py
@@ -239,10 +239,10 @@ class StateStore(SQLBaseStore):
             for event_id, group in event_to_groups.items()
         }
 
-        defer.returnValue([
-            event_to_state[event]
+        defer.returnValue({
+            event: event_to_state[event]
             for event in event_ids
-        ])
+        })
 
     @cached(num_args=2, lru=True, max_entries=100000)
     def _get_state_group_for_event(self, room_id, event_id):