diff --git a/libagent/ssh/__init__.py b/libagent/ssh/__init__.py
index df3cb7d..caf3e18 100644
--- a/libagent/ssh/__init__.py
+++ b/libagent/ssh/__init__.py
@@ -135,7 +135,7 @@ def handle_connection_error(func):
     def wrapper(*args, **kwargs):
         try:
             return func(*args, **kwargs)
-        except IOError as e:
+        except device.interface.NotFoundError as e:
             log.error('Connection error (try unplugging and replugging your device): %s', e)
             return 1
     return wrapper