diff --git a/tests/test.py b/tests/test.py
index c0c8bc0a199ad6227c99e97faf9dcb62fc757766..c2efdef7e12f3c37e373bec564deba013c587e66 100644
--- a/tests/test.py
+++ b/tests/test.py
@@ -4,7 +4,7 @@
 import unittest
 
 from you_get import *
-from you_get.main import url_to_module
+from you_get.__main__ import url_to_module
 
 class YouGetTests(unittest.TestCase):