From 6b7122551a33d698f17c4a960fcab1302b77a237 Mon Sep 17 00:00:00 2001
From: Siming Yuan <siyuan@cisco.com>
Date: Mon, 18 Jul 2016 14:00:59 -0400
Subject: [PATCH] fix for iosxe connection

---
 ncclient/devices/iosxe.py | 4 +++-
 1 file changed, 3 insertions(+), 1 deletion(-)

diff --git a/ncclient/devices/iosxe.py b/ncclient/devices/iosxe.py
index dbd24433..b85445ee 100644
--- a/ncclient/devices/iosxe.py
+++ b/ncclient/devices/iosxe.py
@@ -15,6 +15,8 @@
 
 from .default import DefaultDeviceHandler
 
+from ncclient.operations.third_party.iosxe.rpc import SaveConfig
+
 def iosxe_unknown_host_cb(host, fingerprint):
         #This will ignore the unknown host check when connecting to CSR devices
         return True
@@ -35,7 +37,7 @@ def add_additional_operations(self):
     def add_additional_ssh_connect_params(self, kwargs):
         kwargs['allow_agent']   = False
         kwargs['look_for_keys'] = False
-        kwargs['unknown_host_cb'] = csr_unknown_host_cb
+        kwargs['unknown_host_cb'] = iosxe_unknown_host_cb
 
     def perform_qualify_check(self):
         return False