From b099a1db1992a7ba5b0f3526b7c73ee1e17b2a86 Mon Sep 17 00:00:00 2001
From: Arne Binder <arne.binder@dfki.de>
Date: Tue, 17 Sep 2024 17:37:21 +0200
Subject: [PATCH] fix method call

---
 .../taskmodules/common/taskmodule_with_document_converter.py  | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/src/pie_modules/taskmodules/common/taskmodule_with_document_converter.py b/src/pie_modules/taskmodules/common/taskmodule_with_document_converter.py
index 89e581ed9..267b144f9 100644
--- a/src/pie_modules/taskmodules/common/taskmodule_with_document_converter.py
+++ b/src/pie_modules/taskmodules/common/taskmodule_with_document_converter.py
@@ -89,8 +89,8 @@ def decode(self, **kwargs) -> Sequence[DocumentType]:
         result = []
         for doc in decoded_documents:
             original_document = doc.metadata["original_document"]
-            self._integrate_predictions_into_original_document(
-                document=doc, original_document=original_document
+            self._integrate_predictions_from_converted_document(
+                converted_document=doc, document=original_document
             )
             result.append(original_document)
         return result