diff --git a/telethon/sessions/memory.py b/telethon/sessions/memory.py index 721b6d9a..5aed6039 100644 --- a/telethon/sessions/memory.py +++ b/telethon/sessions/memory.py @@ -174,7 +174,7 @@ class MemorySession(Session): def get_entity_rows_by_id(self, id, exact=True): try: if exact: - return next((id, hash) for found_id, hash, _, _, _ + return next((found_id, hash) for found_id, hash, _, _, _ in self._entities if found_id == id) else: ids = ( @@ -182,7 +182,7 @@ class MemorySession(Session): utils.get_peer_id(PeerChat(id)), utils.get_peer_id(PeerChannel(id)) ) - return next((id, hash) for found_id, hash, _, _, _ + return next((found_id, hash) for found_id, hash, _, _, _ in self._entities if found_id in ids) except StopIteration: pass