diff --git a/changelogs/fragments/check_mode.yaml b/changelogs/fragments/check_mode.yaml
new file mode 100644
index 0000000..5a88245
--- /dev/null
+++ b/changelogs/fragments/check_mode.yaml
@@ -0,0 +1,3 @@
+---
+trivial:
+  - check_mode has been removed from play_context, set on task instead for unit tests
diff --git a/docs/ansible.utils.index_of_lookup.rst b/docs/ansible.utils.index_of_lookup.rst
index 13200ba..fd0ed39 100644
--- a/docs/ansible.utils.index_of_lookup.rst
+++ b/docs/ansible.utils.index_of_lookup.rst
@@ -44,6 +44,7 @@ Parameters
                     <a class="ansibleOptionLink" href="#parameter-" title="Permalink to this option"></a>
                     <div style="font-size: small">
                         <span style="color: purple">list</span>
+                         / <span style="color: purple">elements=raw</span>
                          / <span style="color: red">required</span>
                     </div>
                 </td>
diff --git a/tests/unit/plugins/action/test_fact_diff.py b/tests/unit/plugins/action/test_fact_diff.py
index 402dfaa..984c832 100644
--- a/tests/unit/plugins/action/test_fact_diff.py
+++ b/tests/unit/plugins/action/test_fact_diff.py
@@ -25,7 +25,10 @@ except ImportError:
 class TestUpdate_Fact(unittest.TestCase):
     def setUp(self):
         task = MagicMock(Task)
+        # Ansible > 2.13 looks for check_mode in task
+        task.check_mode = False
         play_context = MagicMock()
+        # Ansible <= 2.13 looks for check_mode in play_context
         play_context.check_mode = False
         connection = MagicMock()
         fake_loader = {}
diff --git a/tests/unit/plugins/action/test_update_fact.py b/tests/unit/plugins/action/test_update_fact.py
index 9eeb11b..c9d395d 100644
--- a/tests/unit/plugins/action/test_update_fact.py
+++ b/tests/unit/plugins/action/test_update_fact.py
@@ -81,7 +81,10 @@ INVALID_JINJA = [
 class TestUpdate_Fact(unittest.TestCase):
     def setUp(self):
         task = MagicMock(Task)
+        # Ansible > 2.13 looks for check_mode in task
+        task.check_mode = False
         play_context = MagicMock()
+        # Ansible <= 2.13 looks for check_mode in play_context
         play_context.check_mode = False
         connection = MagicMock()
         fake_loader = {}