diff --git a/src/test/java/zw/qantra/tm/domain/services/processors/workflows/handlers/GatewayPaymentHandlerTest.java b/src/test/java/zw/qantra/tm/domain/services/processors/workflows/handlers/GatewayPaymentHandlerTest.java
index 0d92b27..319f177 100644
--- a/src/test/java/zw/qantra/tm/domain/services/processors/workflows/handlers/GatewayPaymentHandlerTest.java
+++ b/src/test/java/zw/qantra/tm/domain/services/processors/workflows/handlers/GatewayPaymentHandlerTest.java
@@ -79,6 +79,9 @@ class GatewayPaymentHandlerTest {
         when(transactionProcessor.process(any(Transaction.class))).thenAnswer(invocation -> {
             Transaction tx = invocation.getArgument(0);
             tx.setStatus(Status.SUCCESS);
+            tx.setPaymentStatus(Status.SUCCESS);
+            tx.setPollingStatus(Status.PENDING);
+            tx.setStatus(Status.SUCCESS);
             return tx;
         });