diff --git a/lib/ansible/plugins/lookup/filetree.py b/lib/ansible/plugins/lookup/filetree.py
index 7ff5621214a..3cc6796f0cd 100644
--- a/lib/ansible/plugins/lookup/filetree.py
+++ b/lib/ansible/plugins/lookup/filetree.py
@@ -32,8 +32,11 @@ except ImportError:
 from ansible.plugins.lookup import LookupBase
 from ansible.module_utils._text import to_native
 
-from __main__ import display
-
+try:
+    from __main__ import display
+except ImportError:
+    from ansible.utils.display import Display
+    display = Display()
 
 # If selinux fails to find a default, return an array of None
 def selinux_context(path):