diff --git a/telethon/telegram_client.py b/telethon/telegram_client.py
index 11037257..1575be5d 100644
--- a/telethon/telegram_client.py
+++ b/telethon/telegram_client.py
@@ -930,7 +930,7 @@ class TelegramClient(TelegramBareClient):
         """
         try:
             # First try to get the entity from cache, otherwise figure it out
-            self.session.entities.get_input_entity(peer)
+            return self.session.entities.get_input_entity(peer)
         except KeyError:
             pass
 
@@ -960,7 +960,7 @@ class TelegramClient(TelegramBareClient):
             self.get_dialogs(limit=None)
 
             try:
-                self.session.entities.get_input_entity(peer)
+                return self.session.entities.get_input_entity(peer)
             except KeyError:
                 pass