diff --git a/xmpp_tg/message_handlers.py b/xmpp_tg/message_handlers.py
index dcb6aec..402f657 100644
--- a/xmpp_tg/message_handlers.py
+++ b/xmpp_tg/message_handlers.py
@@ -2,6 +2,7 @@ import time
 import logging, pprint
 
 from xmpp_tg.utils import var_dump, display_tg_name, get_contact_jid, localtime
+from telethon.tl.types import User, Chat, Channel
 
 # modified by adnidor