1 files changed, 7 insertions, 1 deletions
diff --git a/test/units/executor/module_common/test_modify_module.py b/test/units/executor/module_common/test_modify_module.py
index 89e4a163..dceef763 100644
--- a/test/units/executor/module_common/test_modify_module.py
+++ b/test/units/executor/module_common/test_modify_module.py
@@ -8,6 +8,9 @@ __metaclass__ = type
 import pytest
 
 from ansible.executor.module_common import modify_module
+from ansible.module_utils.six import PY2
+
+from test_module_common import templar
 
 
 FAKE_OLD_MODULE = b'''#!/usr/bin/python
@@ -19,7 +22,10 @@ print('{"result": "%s"}' % sys.executable)
 @pytest.fixture
 def fake_old_module_open(mocker):
     m = mocker.mock_open(read_data=FAKE_OLD_MODULE)
-    mocker.patch('builtins.open', m)
+    if PY2:
+        mocker.patch('__builtin__.open', m)
+    else:
+        mocker.patch('builtins.open', m)
 
 # this test no longer makes sense, since a Python module will always either have interpreter discovery run or
 # an explicit interpreter passed (so we'll never default to the module shebang)