From 42d3b467fbec32b1314042271c2a08c798ce025f Mon Sep 17 00:00:00 2001
From: Dan <14043624+delivrance@users.noreply.github.com>
Date: Fri, 10 Aug 2018 11:17:21 +0200
Subject: [PATCH] Fix init not having message and media types

---
 pyrogram/client/types/__init__.py | 8 +++-----
 1 file changed, 3 insertions(+), 5 deletions(-)

diff --git a/pyrogram/client/types/__init__.py b/pyrogram/client/types/__init__.py
index 3fd4a8d1..230d5e5d 100644
--- a/pyrogram/client/types/__init__.py
+++ b/pyrogram/client/types/__init__.py
@@ -28,13 +28,11 @@ from .input_media import (
     InputMediaAudio, InputPhoneContact, InputMediaVideo, InputMediaPhoto,
     InputMediaDocument, InputMediaAnimation
 )
-from .media import (
+from .messages_and_media import (
     Audio, Contact, Document, Animation, Location, Photo, PhotoSize,
-    Sticker, Venue, Video, VideoNote, Voice, UserProfilePhotos
+    Sticker, Venue, Video, VideoNote, Voice, UserProfilePhotos,
+    Message, Messages, MessageEntity
 )
-from .message import Message
-from .message_entity import MessageEntity
-from .messages import Messages
 from .update import Update
 from .user_and_chats import (
     Chat, ChatMember, ChatMembers, ChatPhoto,