From c788e17c2275ddfbd4c29c08d909da660a46cf1f Mon Sep 17 00:00:00 2001
From: Lonami Exo <totufals@hotmail.com>
Date: Thu, 14 Jun 2018 22:51:57 +0200
Subject: [PATCH] Fix _handle_update not expanding Updates

---
 telethon/client/updates.py | 12 ++++++++++++
 1 file changed, 12 insertions(+)

diff --git a/telethon/client/updates.py b/telethon/client/updates.py
index 9167e40e..8d7adef9 100644
--- a/telethon/client/updates.py
+++ b/telethon/client/updates.py
@@ -1,4 +1,5 @@
 import asyncio
+import itertools
 import logging
 import warnings
 
@@ -144,6 +145,17 @@ class UpdateMethods(UserMethods):
     # region Private methods
 
     def _handle_update(self, update):
+        if isinstance(update, (types.Updates, types.UpdatesCombined)):
+            entities = {utils.get_peer_id(x): x for x in
+                        itertools.chain(update.users, update.chats)}
+            for u in update.updates:
+                u._entities = entities
+                asyncio.ensure_future(
+                    self._dispatch_update(u), loop=self._loop)
+            return
+        if isinstance(update, types.UpdateShort):
+            update = update.update
+        update._entities = {}
         asyncio.ensure_future(self._dispatch_update(update), loop=self._loop)
 
     async def _dispatch_update(self, update):