From b7c0c2ab10a5b494e72fc624e0d462691192bb05 Mon Sep 17 00:00:00 2001
From: Isobelle Lim <isobelle.lim@klaviyo.com>
Date: Thu, 19 Dec 2024 15:04:28 -0500
Subject: [PATCH] fix failing test

---
 Tests/KlaviyoSwiftTests/KlaviyoSDKTests.swift | 3 ++-
 1 file changed, 2 insertions(+), 1 deletion(-)

diff --git a/Tests/KlaviyoSwiftTests/KlaviyoSDKTests.swift b/Tests/KlaviyoSwiftTests/KlaviyoSDKTests.swift
index 3a237063..92eae7f7 100644
--- a/Tests/KlaviyoSwiftTests/KlaviyoSDKTests.swift
+++ b/Tests/KlaviyoSwiftTests/KlaviyoSDKTests.swift
@@ -146,6 +146,7 @@ class KlaviyoSDKTests: XCTestCase {
     // MARK: test unhandle push notification
 
     func testUnhandlePushNotification() throws {
+        let expectation = setupActionAssertion(expectedAction: .syncBadgeCount)
         let callback = XCTestExpectation(description: "callback is not made")
         callback.isInverted = true
         let data: [AnyHashable: Any] = [
@@ -161,7 +162,7 @@ class KlaviyoSDKTests: XCTestCase {
             callback.fulfill()
         }
 
-        wait(for: [callback], timeout: 1.0)
+        wait(for: [callback, expectation], timeout: 1.0)
         XCTAssertFalse(handled)
     }