From 4bf0e4be5cb3539a75e19c8ec7b258916768bdf9 Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?Damir=20Jeli=C4=87?= <poljar@termina.org.uk>
Date: Sat, 13 Oct 2018 12:14:54 +0200
Subject: [PATCH] server: Mark a finished key query.

---
 matrix/server.py | 4 ++++
 1 file changed, 4 insertions(+)

diff --git a/matrix/server.py b/matrix/server.py
index 8e6c7c2..a4e4690 100644
--- a/matrix/server.py
+++ b/matrix/server.py
@@ -34,6 +34,7 @@ from nio import (
     SyncResponse,
     PartialSyncResponse,
     ShareGroupSessionResponse,
+    KeysQueryResponse,
     KeysClaimResponse,
     TransportResponse,
     TransportType,
@@ -898,6 +899,9 @@ class MatrixServer(object):
         elif isinstance(response, RoomMessagesResponse):
             self.handle_backlog_response(response)
 
+        elif isinstance(response, KeysQueryResponse):
+            self.keys_queried = False
+
         elif isinstance(response, KeysClaimResponse):
             self.keys_claimed[response.room_id] = False
             try: