From 822bace8d32197407df26ae6a7480edf6e00fab7 Mon Sep 17 00:00:00 2001
From: Harshdeep Singh <6162866+harsh62@users.noreply.github.com>
Date: Thu, 26 Jun 2025 15:48:57 -0400
Subject: [PATCH] chore: fix integration tests

---
 .../IdentifyBasicIntegrationTests.swift                     | 6 +++---
 1 file changed, 3 insertions(+), 3 deletions(-)

diff --git a/AmplifyPlugins/Predictions/Tests/PredictionsHostApp/AWSPredictionsPluginIntegrationTests/IdentifyBasicIntegrationTests.swift b/AmplifyPlugins/Predictions/Tests/PredictionsHostApp/AWSPredictionsPluginIntegrationTests/IdentifyBasicIntegrationTests.swift
index 3b8762e00d..969c0b0dc4 100644
--- a/AmplifyPlugins/Predictions/Tests/PredictionsHostApp/AWSPredictionsPluginIntegrationTests/IdentifyBasicIntegrationTests.swift
+++ b/AmplifyPlugins/Predictions/Tests/PredictionsHostApp/AWSPredictionsPluginIntegrationTests/IdentifyBasicIntegrationTests.swift
@@ -225,11 +225,11 @@ class IdentifyBasicIntegrationTests: AWSPredictionsPluginTestBase {
         XCTAssertNotNil(result)
         XCTAssertFalse(result.fullText.isEmpty)
         XCTAssertFalse(result.words.isEmpty)
-        XCTAssertEqual(result.words.count, 30)
+        XCTAssertEqual(result.words.count, 28)
         XCTAssertFalse(result.rawLineText.isEmpty)
-        XCTAssertEqual(result.rawLineText.count, 17)
+        XCTAssertEqual(result.rawLineText.count, 11)
         XCTAssertFalse(result.identifiedLines.isEmpty)
-        XCTAssertEqual(result.identifiedLines.count, 17)
+        XCTAssertEqual(result.identifiedLines.count, 11)
         XCTAssertFalse(result.keyValues.isEmpty)
         XCTAssertEqual(result.keyValues.count, 7)
     }