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)
     }