Remove suffix from dispatcher_nb

3 files changed, 7 insertions, 7 deletions

diff --git a/test/broken/test_xmpp_smacks.py b/test/broken/test_xmpp_smacks.py
index fa66065..4ccd704 100644
--- a/test/broken/test_xmpp_smacks.py
+++ b/test/broken/test_xmpp_smacks.py
@@ -4,7 +4,7 @@ Tests for smacks.py Stream Management
 import unittest
 from unittest.mock import Mock
 
-from nbxmpp import dispatcher_nb
+from nbxmpp import dispatcher
 from nbxmpp import protocol
 from nbxmpp import smacks
 
@@ -14,7 +14,7 @@ class TestDispatcherNB(unittest.TestCase):
     into a mock client
     '''
     def setUp(self):
-        self.dispatcher = dispatcher_nb.XMPPDispatcher()
+        self.dispatcher = dispatcher.XMPPDispatcher()
 
         # Setup mock client
         self.client = Mock()
diff --git a/test/unit/test_xml_vulnerability.py b/test/unit/test_xml_vulnerability.py
index 7a8cb7a..dcb0a7d 100644
--- a/test/unit/test_xml_vulnerability.py
+++ b/test/unit/test_xml_vulnerability.py
@@ -1,7 +1,7 @@
 import unittest
 from unittest.mock import Mock
 
-from nbxmpp import dispatcher_nb
+from nbxmpp import dispatcher
 
 
 class XMLVulnerability(unittest.TestCase):
@@ -9,7 +9,7 @@ class XMLVulnerability(unittest.TestCase):
     def setUp(self):
         self.client = Mock()
         self.client.Connection = Mock()
-        self.dispatcher = dispatcher_nb.XMPPDispatcher()
+        self.dispatcher = dispatcher.XMPPDispatcher()
         self.dispatcher.PlugIn(self.client)
         self.dispatcher.StreamInit()
 
diff --git a/test/unit/test_xmpp_dispatcher_nb.py b/test/unit/test_xmpp_dispatcher.py
index 8c4e49b..d8ac05e 100644
--- a/test/unit/test_xmpp_dispatcher_nb.py
+++ b/test/unit/test_xmpp_dispatcher.py
@@ -1,12 +1,12 @@
 '''
-Tests for dispatcher_nb.py
+Tests for dispatcher.py
 '''
 import unittest
 from unittest.mock import Mock
 
 from test import lib
 
-from nbxmpp import dispatcher_nb
+from nbxmpp import dispatcher
 from nbxmpp import protocol
 from nbxmpp.protocol import JID
 
@@ -17,7 +17,7 @@ class TestDispatcherNB(unittest.TestCase):
     into a mock client
     '''
     def setUp(self):
-        self.dispatcher = dispatcher_nb.XMPPDispatcher()
+        self.dispatcher = dispatcher.XMPPDispatcher()
 
         # Setup mock client
         self.client = Mock()