diff --git a/test/units/mock/loader.py b/test/units/mock/loader.py
index 8b6bbbbaf9..f44df2efdb 100644
--- a/test/units/mock/loader.py
+++ b/test/units/mock/loader.py
@@ -21,6 +21,7 @@ __metaclass__ = type
 
 import os
 
+from ansible.errors import AnsibleParserError
 from ansible.parsing import DataLoader
 
 class DictDataLoader(DataLoader):