Take value in a better way

2 files changed, 3 insertions, 3 deletions

diff --git a/synapse/storage/events.py b/synapse/storage/events.py

index bec35ea68d..ed182c8d11 100644
--- a/synapse/storage/events.py
+++ b/synapse/storage/events.py
@@ -1616,7 +1616,7 @@ class EventsStore(SQLBaseStore):
             curr_state = self._get_state_groups_from_groups_txn(
                 txn, [new_state_edge], types=None
             )
-            curr_state = curr_state.values()[0]
+            curr_state = curr_state[new_state_edge]
 
             self._simple_delete_txn(
                 txn,
diff --git a/synapse/storage/state.py b/synapse/storage/state.py

index e790793370..589a4fec6e 100644
--- a/synapse/storage/state.py
+++ b/synapse/storage/state.py
@@ -693,12 +693,12 @@ class StateStore(SQLBaseStore):
                     prev_state = self._get_state_groups_from_groups_txn(
                         txn, [prev_group], types=None
                     )
-                    prev_state = prev_state.values()[0]
+                    prev_state = prev_state[prev_group]
 
                     curr_state = self._get_state_groups_from_groups_txn(
                         txn, [state_group], types=None
                     )
-                    curr_state = curr_state.values()[0]
+                    curr_state = curr_state[state_group]
 
                     if not set(prev_state.keys()) - set(curr_state.keys()):
                         # We can only do a delta if the current has a strict super set