diff --git a/src/main/java/zw/qantra/tm/domain/workflows/TransactionWorkflow.java b/src/main/java/zw/qantra/tm/domain/workflows/TransactionWorkflow.java
index 5c33753..1d6abcd 100644
--- a/src/main/java/zw/qantra/tm/domain/workflows/TransactionWorkflow.java
+++ b/src/main/java/zw/qantra/tm/domain/workflows/TransactionWorkflow.java
@@ -236,9 +236,12 @@ public class TransactionWorkflow extends WorkflowDefinition {
             transaction = process(transaction, "integration");
             execution.setVariable("body", Utils.toJson(transaction));
             return NextAction.moveToState(PURCHASE_INVOICE_PAYMENT, "Integration request successful");
-        }catch (Exception e) {
-            log.info(e.getCause().getMessage());
-            return NextAction.moveToState(FAILED, e.getCause().getMessage());
+        } catch (ApiException e) {
+            log.info(e.getMessage());
+            return NextAction.moveToState(FAILED, e.getMessage());
+        } catch (Exception e) {
+            log.info(e.getMessage());
+            return NextAction.moveToState(FAILED, e.getMessage());
         }
     }
 
diff --git a/src/main/java/zw/qantra/tm/domain/workflows/handlers/IntegrationHandler.java b/src/main/java/zw/qantra/tm/domain/workflows/handlers/IntegrationHandler.java
index 16d70c2..f3ccb9a 100644
--- a/src/main/java/zw/qantra/tm/domain/workflows/handlers/IntegrationHandler.java
+++ b/src/main/java/zw/qantra/tm/domain/workflows/handlers/IntegrationHandler.java
@@ -12,7 +12,6 @@ import zw.qantra.tm.domain.models.TransactionEvent;
 import zw.qantra.tm.domain.services.*;
 import zw.qantra.tm.domain.services.factories.PaymentProcessorFactory;
 import zw.qantra.tm.domain.services.processors.TransactionProcessorInterface;
-import zw.qantra.tm.domain.services.processors.handlers.LogicPipeline;
 import zw.qantra.tm.exceptions.ApiException;
 import zw.qantra.tm.utils.Utils;