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
|
