diff --git a/test/unit-tests/aws/cloud-formation-wrapper.test.ts b/test/unit-tests/aws/cloud-formation-wrapper.test.ts
index a42fc7cd..8afed875 100644
--- a/test/unit-tests/aws/cloud-formation-wrapper.test.ts
+++ b/test/unit-tests/aws/cloud-formation-wrapper.test.ts
@@ -107,6 +107,7 @@ describe("Cloud Formation wrapper checks", () => {
     } catch (err) {
       errored = true;
       expect(err.message).to.contains("Failed to find logicalResourceId");
+      expect(err.message).to.contains("Make sure the stack exists and the API gateway event is added");
     }
     expect(errored).to.equal(true);
 
@@ -160,6 +161,7 @@ describe("Cloud Formation wrapper checks", () => {
     } catch (err) {
       errored = true;
       expect(err.message).to.contains("Failed to find logicalResourceId");
+      expect(err.message).to.contains("Make sure the stack exists and the API gateway event is added");
     }
     expect(errored).to.equal(true);
 
@@ -185,6 +187,7 @@ describe("Cloud Formation wrapper checks", () => {
     } catch (err) {
       errored = true;
       expect(err.message).to.contains("Failed to find logicalResourceId");
+      expect(err.message).to.contains("Make sure the stack exists and the API gateway event is added");
     }
     expect(errored).to.equal(true);
     expect(consoleOutput[0]).to.contains("Unsupported apiGateway");
@@ -389,6 +392,7 @@ describe("Cloud Formation wrapper checks", () => {
     } catch (err) {
       errored = true;
       expect(err.message).to.contains("Failed to find logicalResourceId");
+      expect(err.message).to.contains("Make sure the stack exists and the API gateway event is added");
     }
     expect(errored).to.equal(true);
     expect(consoleOutput[0]).to.contains("[WARNING] Failed to find CloudFormation resources with an error");