diff --git a/synapse/storage/state.py b/synapse/storage/state.py
index abc5b6643..1f5e62fa8 100644
--- a/synapse/storage/state.py
+++ b/synapse/storage/state.py
@@ -426,14 +426,9 @@ class StateStore(SQLBaseStore):
             else:
                 state_dict = results[group]
 
-            evs = [
-                state_events[e_id] for e_id in state_ids
-                if e_id in state_events  # This can happen if event is rejected.
-            ]
-            state_dict.update({
-                (e.type, e.state_key): e
-                for e in evs
-            })
+            for event_id in state_ids:
+                state_event = state_events[event_id]
+                state_dict[(state_event.type, state_event.state_key)] = state_event
 
             self._state_group_cache.update(
                 cache_seq_num,