diff --git a/tests/echo.py b/tests/echo.py
index 878c31a..dd35810 100755
--- a/tests/echo.py
+++ b/tests/echo.py
@@ -12,7 +12,7 @@ as taken from http://docs.python.org/dev/library/ssl.html#certificates
 
 import os, sys, select, optparse
 sys.path.insert(0,os.path.dirname(__file__) + "/../")
-from websocket import WebSocketServer
+from websockify.websocket import WebSocketServer
 
 class WebSocketEcho(WebSocketServer):
     """
diff --git a/tests/load.py b/tests/load.py
index 0501f7d..b298703 100755
--- a/tests/load.py
+++ b/tests/load.py
@@ -8,7 +8,7 @@ given a sequence number. Any errors are reported and counted.
 
 import sys, os, select, random, time, optparse
 sys.path.insert(0,os.path.dirname(__file__) + "/../")
-from websocket import WebSocketServer
+from websockify.websocket import WebSocketServer
 
 class WebSocketLoad(WebSocketServer):
 
diff --git a/tests/utf8-list.py b/tests/utf8-list.py
index 5a36da0..1ae1238 100755
--- a/tests/utf8-list.py
+++ b/tests/utf8-list.py
@@ -7,7 +7,7 @@ import sys, os, socket, ssl, time, traceback
 from select import select
 
 sys.path.insert(0,os.path.dirname(__file__) + "/../")
-from websocket import WebSocketServer
+from websockify.websocket import WebSocketServer
 
 if __name__ == '__main__':
     print "val: hixie | hybi_base64 | hybi_binary"