diff --git a/ds-caselaw-ingester/tests.py b/ds-caselaw-ingester/tests.py
index 1f9f4de..49374b9 100644
--- a/ds-caselaw-ingester/tests.py
+++ b/ds-caselaw-ingester/tests.py
@@ -650,7 +650,7 @@ def test_update_document_xml_success(self, api_client):
         xml = ET.XML("<xml>Here's some xml</xml>")
         api_client.get_judgment_xml = MagicMock(return_value=True)
         api_client.update_document_xml = MagicMock(return_value=True)
-        result = lambda_function.update_document_xml("a/fake/uri", xml)
+        result = lambda_function.update_document_xml("a/fake/uri", xml, {"test": True})
         assert result is True
 
     @patch("lambda_function.api_client", autospec=True)
@@ -660,7 +660,7 @@ def test_update_document_xml_judgment_does_not_exist(self, api_client):
             side_effect=MarklogicResourceNotFoundError("error")
         )
         api_client.update_document_xml = MagicMock(return_value=True)
-        result = lambda_function.update_document_xml("a/fake/uri", xml)
+        result = lambda_function.update_document_xml("a/fake/uri", xml, {"test": True})
         assert result is False
 
     @patch("lambda_function.api_client", autospec=True)
@@ -671,13 +671,13 @@ def test_update_document_xml_judgment_does_not_save(self, api_client):
             side_effect=MarklogicCommunicationError("error")
         )
         with pytest.raises(MarklogicCommunicationError):
-            lambda_function.update_document_xml("a/fake/uri", xml)
+            lambda_function.update_document_xml("a/fake/uri", xml, {"test": True})
 
     @patch("lambda_function.api_client", autospec=True)
     def test_insert_document_xml_success(self, api_client):
         xml = ET.XML("<xml>Here's some xml</xml>")
         api_client.insert_document_xml = MagicMock(return_value=True)
-        result = lambda_function.insert_document_xml("a/fake/uri", xml)
+        result = lambda_function.insert_document_xml("a/fake/uri", xml, {"test": True})
         assert result is True
 
     @patch("lambda_function.api_client", autospec=True)
@@ -687,7 +687,7 @@ def test_insert_document_xml_failure(self, api_client):
             side_effect=MarklogicCommunicationError("error")
         )
         with pytest.raises(MarklogicCommunicationError):
-            lambda_function.insert_document_xml("a/fake/uri", xml)
+            lambda_function.insert_document_xml("a/fake/uri", xml, {"test": True})
 
     def test_get_best_xml_with_valid_xml_file(self):
         filename = "TDR-2022-DNWR.xml"