diff --git a/telethon/telegram_client.py b/telethon/telegram_client.py
index c1eab9fa..3a264b42 100644
--- a/telethon/telegram_client.py
+++ b/telethon/telegram_client.py
@@ -1079,7 +1079,7 @@ class TelegramClient(TelegramBareClient):
         try:
             # First try to get the entity from cache, otherwise figure it out
             return self.session.get_input_entity(peer)
-        except KeyError:
+        except ValueError:
             pass
 
         if isinstance(peer, str):