diff --git a/telethon_generator/parser/tl_object.py b/telethon_generator/parser/tl_object.py
index 278a66eb..034cb3c3 100644
--- a/telethon_generator/parser/tl_object.py
+++ b/telethon_generator/parser/tl_object.py
@@ -264,7 +264,7 @@ class TLArg:
             'date': 'datetime.datetime | None',  # None date = 0 timestamp
             'bytes': 'bytes',
             'true': 'bool',
-        }.get(self.type, 'TLObject')
+        }.get(self.type, self.type)
         if self.is_vector:
             result = 'list[{}]'.format(result)
         if self.is_flag and self.type != 'date':
diff --git a/telethon_generator/tl_generator.py b/telethon_generator/tl_generator.py
index fb8ca4bd..18293ba1 100644
--- a/telethon_generator/tl_generator.py
+++ b/telethon_generator/tl_generator.py
@@ -146,15 +146,6 @@ class TLGenerator:
                         x for x in namespace_tlobjects.keys() if x
                     )))
 
-                # Import 'get_input_*' utils
-                # TODO Support them on types too
-                if 'functions' in out_dir:
-                    builder.writeln(
-                        'from {}.utils import get_input_peer, '
-                        'get_input_channel, get_input_user, '
-                        'get_input_media, get_input_photo'.format('.' * depth)
-                    )
-
                 # Import 'os' for those needing access to 'os.urandom()'
                 # Currently only 'random_id' needs 'os' to be imported,
                 # for all those TLObjects with arg.can_be_inferred.