ssh: fix exception type for missing device

7 years ago · 1942e3999b
parent c4dfca04f2
commit 1942e3999b
1 changed files with 1 additions and 1 deletions
--- 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