From ad26ac514962b71da83925e04122eb4c311a57d6 Mon Sep 17 00:00:00 2001
From: Meher Mankikar <mmankika@amazon.com>
Date: Mon, 19 Jun 2023 16:25:01 -0700
Subject: [PATCH] Bug fix from last revision

---
 .../amazon/kinesis/utils/StreamExistenceManager.java      | 8 +++++---
 1 file changed, 5 insertions(+), 3 deletions(-)

diff --git a/amazon-kinesis-client/src/test/java/software/amazon/kinesis/utils/StreamExistenceManager.java b/amazon-kinesis-client/src/test/java/software/amazon/kinesis/utils/StreamExistenceManager.java
index 0181a222..ee0450a0 100644
--- a/amazon-kinesis-client/src/test/java/software/amazon/kinesis/utils/StreamExistenceManager.java
+++ b/amazon-kinesis-client/src/test/java/software/amazon/kinesis/utils/StreamExistenceManager.java
@@ -43,10 +43,12 @@ public class StreamExistenceManager {
                 throw new RuntimeException("Stream is not active, instead in status: " + response.streamDescriptionSummary().streamStatus());
             }
             return true;
-        } catch (ResourceNotFoundException e) {
-            return false;
         } catch (ExecutionException e) {
-            throw new RuntimeException(e);
+            if (e.getCause() instanceof ResourceNotFoundException) {
+                return false;
+            } else {
+                throw new RuntimeException(e);
+            }
         } catch (Exception e) {
             throw new RuntimeException(e);
         }