From 1942e3999b40fe1c5c69f3645684b9508178d21f Mon Sep 17 00:00:00 2001
From: Roman Zeyde <roman.zeyde@gmail.com>
Date: Sat, 18 Nov 2017 16:50:08 +0200
Subject: [PATCH] ssh: fix exception type for missing device

---
 libagent/ssh/__init__.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

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