1 files changed, 2 insertions, 1 deletions
diff --git a/test/test_log.py b/test/test_log.py
index 2cac1e7c..8f38c040 100644
--- a/test/test_log.py
+++ b/test/test_log.py
@@ -1,5 +1,6 @@
 import StringIO
 from libpathod import log
+from netlib.exceptions import TcpDisconnect
 import netlib.tcp
 
 
@@ -19,6 +20,6 @@ def test_disconnect():
     try:
         with l.ctx() as lg:
             lg("Test")
-    except netlib.tcp.NetLibDisconnect:
+    except TcpDisconnect:
         pass
     assert "Test" in outf.getvalue()