fixing imports

4 files changed, 4 insertions, 4 deletions

diff --git a/python-tornado/tests/benchmark.py b/python-tornado/tests/benchmark.py
index 636a87a..748fe3b 100644
--- a/python-tornado/tests/benchmark.py
+++ b/python-tornado/tests/benchmark.py
@@ -12,7 +12,7 @@
 import sys
 import datetime
 import tornado
-from Pubnub import Pubnub
+from Pubnub import PubnubTwisted as Pubnub
 
 publish_key   = len(sys.argv) > 1 and sys.argv[1] or 'demo'
 subscribe_key = len(sys.argv) > 2 and sys.argv[2] or 'demo'
diff --git a/python-tornado/tests/delivery.py b/python-tornado/tests/delivery.py
index d0eb9f9..0181403 100644
--- a/python-tornado/tests/delivery.py
+++ b/python-tornado/tests/delivery.py
@@ -14,7 +14,7 @@ import datetime
 import time
 import math
 
-from Pubnub import Pubnub
+from Pubnub import PubnubTwisted as Pubnub
 
 ## -----------------------------------------------------------------------
 ## Configuration
diff --git a/python-tornado/tests/subscribe-test.py b/python-tornado/tests/subscribe-test.py
index a2bafde..bcbbc7e 100755
--- a/python-tornado/tests/subscribe-test.py
+++ b/python-tornado/tests/subscribe-test.py
@@ -11,7 +11,7 @@
 
 import sys
 import datetime
-from Pubnub import Pubnub
+from Pubnub import PubnubTwisted as Pubnub
 from functools import partial
 from threading import current_thread
 import threading
diff --git a/python-tornado/tests/unit-tests.py b/python-tornado/tests/unit-tests.py
index 0069a5c..b00f2b7 100644
--- a/python-tornado/tests/unit-tests.py
+++ b/python-tornado/tests/unit-tests.py
@@ -3,7 +3,7 @@ import sys
 
 
 from PubnubUnitTest import Suite
-from Pubnub import Pubnub
+from Pubnub import PubnubTwisted as Pubnub
 
 pubnub = Pubnub("demo", "demo")