diff --git a/websockify/__init__.py b/websockify/__init__.py
index e8f2eaf..37a6f47 100644
--- a/websockify/__init__.py
+++ b/websockify/__init__.py
@@ -1,2 +1,2 @@
-from websocket import *
-from websocketproxy import *
+from websockify.websocket import *
+from websockify.websocketproxy import *
diff --git a/websockify/websocketproxy.py b/websockify/websocketproxy.py
index 17d9451..ac9ed63 100755
--- a/websockify/websocketproxy.py
+++ b/websockify/websocketproxy.py
@@ -17,7 +17,7 @@ except: from SocketServer import ForkingMixIn
 try:    from http.server import HTTPServer
 except: from BaseHTTPServer import HTTPServer
 from select import select
-import websocket
+from websockify import websocket
 try:
     from urllib.parse import parse_qs, urlparse
 except: