diff --git a/persistence-service-v1/persistence-service-server-v1/src/test/java/com/iqser/red/service/peristence/v1/server/integration/utils/AbstractPersistenceServerServiceTest.java b/persistence-service-v1/persistence-service-server-v1/src/test/java/com/iqser/red/service/peristence/v1/server/integration/utils/AbstractPersistenceServerServiceTest.java
index ce7aa679a..4dd8cd27f 100644
--- a/persistence-service-v1/persistence-service-server-v1/src/test/java/com/iqser/red/service/peristence/v1/server/integration/utils/AbstractPersistenceServerServiceTest.java
+++ b/persistence-service-v1/persistence-service-server-v1/src/test/java/com/iqser/red/service/peristence/v1/server/integration/utils/AbstractPersistenceServerServiceTest.java
@@ -36,6 +36,7 @@ import org.springframework.test.context.junit4.SpringRunner;
 import static org.mockito.Mockito.doNothing;
 import static org.mockito.Mockito.when;
 
+import java.util.List;
 import java.util.TimeZone;
 
 @RunWith(SpringRunner.class)
@@ -114,10 +115,11 @@ public abstract class AbstractPersistenceServerServiceTest {
         doNothing().when(pdfTronRedactionClient).testDigitalCurrentSignature(Mockito.any());
 
         when(amqpAdmin.getQueueInfo(Mockito.any())).thenReturn(null);
-        when(pdfTronRedactionClient.optimize(Mockito.any())).thenAnswer((args) -> {
-            PdfTronOptimizeRequest request = (PdfTronOptimizeRequest) args.getArguments()[0];
+
+        when(pdfTronRedactionClient.optimizeGetAndRemoveAnnotations(Mockito.any())).thenAnswer((args) -> {
+            DocumentRequest request = (DocumentRequest) args.getArguments()[0];
             int numberOfPages = (request.getDocument().length == 0? 0: 10);
-            return new PdfTronOptimizeResponse(request.getDocument(), numberOfPages);
+            return new AnnotationExtractionResponse(request.getDocument(), numberOfPages, List.of());
         });
         when(pdfTronRedactionClient.redact(Mockito.any())).thenAnswer((args) ->
                 new PdfTronRedactionResult(((PdfTronRedactionRequest) args.getArguments()[0]).getDocument()));