From 2628fe1eb0254330f05010e1219b64c7d9201b2f Mon Sep 17 00:00:00 2001
From: Niket Agrawal <niket.agrawal90@gmail.com>
Date: Tue, 7 Nov 2023 23:13:46 +0530
Subject: [PATCH] Review Comment changes

---
 .../FirstMatchingElementUnitTest.java                     | 8 ++++++--
 1 file changed, 6 insertions(+), 2 deletions(-)

diff --git a/core-java-modules/core-java-streams-5/src/test/java/com/baeldung/streams/firstmatchingelement/FirstMatchingElementUnitTest.java b/core-java-modules/core-java-streams-5/src/test/java/com/baeldung/streams/firstmatchingelement/FirstMatchingElementUnitTest.java
index 48171bab65..2021014957 100644
--- a/core-java-modules/core-java-streams-5/src/test/java/com/baeldung/streams/firstmatchingelement/FirstMatchingElementUnitTest.java
+++ b/core-java-modules/core-java-streams-5/src/test/java/com/baeldung/streams/firstmatchingelement/FirstMatchingElementUnitTest.java
@@ -44,9 +44,13 @@ public class FirstMatchingElementUnitTest {
     @Test
     public void whenUsingTakeWhile_thenFindFirstMatchingUserIndex() {
         int lastIndex = userList.size() - 1;
+        AtomicInteger counter = new AtomicInteger(-1);
         int predicateIndex = userList.stream()
-          .takeWhile(user -> !(searchName.equals(user.getUserName())))
-          .mapToInt(userList::indexOf)
+            .takeWhile(user -> {
+                counter.getAndIncrement();
+                return !(searchName.equals(user.getUserName()));
+            })
+          .mapToInt(user -> counter.get())
           .max()
           .orElse(-1);