diff --git a/Whatsapp_Chat_Exporter/__main__.py b/Whatsapp_Chat_Exporter/__main__.py
index d449a91..11ed6c7 100644
--- a/Whatsapp_Chat_Exporter/__main__.py
+++ b/Whatsapp_Chat_Exporter/__main__.py
@@ -1,5 +1,9 @@
-from .__init__ import __version__
-from Whatsapp_Chat_Exporter import extract, extract_iphone
+try:
+    from .__init__ import __version__
+except ImportError:
+    from Whatsapp_Chat_Exporter.__init__ import __version__
+from Whatsapp_Chat_Exporter import extract_new as extract
+from Whatsapp_Chat_Exporter import extract_iphone
 from Whatsapp_Chat_Exporter import extract_iphone_media
 from Whatsapp_Chat_Exporter.extract import Crypt
 from optparse import OptionParser