From 69e113f29d2c85c107006f5804cde1c6b38beec2 Mon Sep 17 00:00:00 2001
From: bencvdb <bencvdb93#gmail.com>
Date: Wed, 7 Jul 2021 09:27:14 -0700
Subject: [PATCH] fix(worker): fix NPE

---
 .../services/kinesis/clientlibrary/lib/worker/Worker.java    | 5 ++---
 1 file changed, 2 insertions(+), 3 deletions(-)

diff --git a/src/main/java/com/fivetran/external/com/amazonaws/services/kinesis/clientlibrary/lib/worker/Worker.java b/src/main/java/com/fivetran/external/com/amazonaws/services/kinesis/clientlibrary/lib/worker/Worker.java
index c72148e0..83af9e2c 100644
--- a/src/main/java/com/fivetran/external/com/amazonaws/services/kinesis/clientlibrary/lib/worker/Worker.java
+++ b/src/main/java/com/fivetran/external/com/amazonaws/services/kinesis/clientlibrary/lib/worker/Worker.java
@@ -574,11 +574,10 @@ public class Worker implements Runnable {
     private <T extends Throwable> Optional<T> getCauseOfType(Class<T> clazz, Throwable t) {
         if (t.getClass().equals(clazz)) {
             return Optional.of( (T) t);
-        }
-        if (t.getCause().getClass().equals(clazz)) {
-            return Optional.of( (T) t.getCause());
         } else if (t.getCause() == null) {
             return Optional.empty();
+        } else if (t.getCause().getClass().equals(clazz)) {
+            return Optional.of( (T) t.getCause());
         }
 
         return getCauseOfType(clazz, t.getCause());