diff --git a/tests/test_connector.py b/tests/test_connector.py
index 36df6f72f08ac0a95f7e7c6f32649b5fd53a3075..40041199cf6cad3df29273c7535d04af081d349e 100644
--- a/tests/test_connector.py
+++ b/tests/test_connector.py
@@ -1,5 +1,5 @@
 import unittest
-from pafsim.connector import Connector, ConnectionError
+from pafsim.connector import Connector, EdgeConnectionError
 import pafsim.processor as pp
 
 class TEST_Connector(unittest.TestCase):
@@ -14,6 +14,6 @@ class TEST_Connector(unittest.TestCase):
         proc1 = pp.Correlator("dummy1")
         proc2 = pp.Correlator("dummy2")
         conn = Connector(proc1, proc2)
-        with self.assertRaises(ConnectionError):
+        with self.assertRaises(EdgeConnectionError):
             conn.connect()