diff --git a/lxc_ssh.py b/lxc_ssh.py
index b425652..3479794 100644
--- a/lxc_ssh.py
+++ b/lxc_ssh.py
@@ -27,6 +27,7 @@
 
 from ansible import constants as C
 from ansible.compat.six import text_type, binary_type
+from ansible.errors import AnsibleError
 from ansible.plugins.connection import ConnectionBase
 from ansible.utils.path import unfrackpath, makedirs_safe
 from ansible.utils.unicode import to_bytes, to_unicode, to_str