diff --git a/telethon_generator/generators/tlobject.py b/telethon_generator/generators/tlobject.py
index 31a47fbe..7b27f60c 100644
--- a/telethon_generator/generators/tlobject.py
+++ b/telethon_generator/generators/tlobject.py
@@ -642,7 +642,7 @@ def _write_patched(out_dir, namespace_tlobjects):
             builder.writeln(AUTO_GEN_NOTICE)
 
             builder.writeln('import struct')
-            builder.writeln('from .. import types, custom')
+            builder.writeln('from .. import TLObject, types, custom')
             builder.writeln()
             for t in tlobjects:
                 builder.writeln('class {}(custom.{}):', t.class_name,