diff --git a/openmetadata-service/src/main/java/org/openmetadata/service/apps/bundles/searchIndex/SearchIndexApp.java b/openmetadata-service/src/main/java/org/openmetadata/service/apps/bundles/searchIndex/SearchIndexApp.java
index 58e940f85f3..e6911298ace 100644
--- a/openmetadata-service/src/main/java/org/openmetadata/service/apps/bundles/searchIndex/SearchIndexApp.java
+++ b/openmetadata-service/src/main/java/org/openmetadata/service/apps/bundles/searchIndex/SearchIndexApp.java
@@ -756,10 +756,9 @@ public class SearchIndexApp extends AbstractNativeApplication {
 
   private void waitForConsumersToComplete(CountDownLatch consumerLatch)
       throws InterruptedException {
-    boolean finished = consumerLatch.await(5, TimeUnit.MINUTES);
-    if (!finished) {
-      LOG.warn("Consumers did not finish within timeout");
-    }
+    LOG.info("Waiting for all consumers to complete their work...");
+    consumerLatch.await(); // Wait indefinitely for consumers to finish all work
+    LOG.info("All consumers have completed their work");
   }
 
   private void handleInterruption(InterruptedException e) throws InterruptedException {
diff --git a/openmetadata-service/src/main/java/org/openmetadata/service/socket/WebSocketManager.java b/openmetadata-service/src/main/java/org/openmetadata/service/socket/WebSocketManager.java
index 6b57fa59d47..e41dc1fd622 100644
--- a/openmetadata-service/src/main/java/org/openmetadata/service/socket/WebSocketManager.java
+++ b/openmetadata-service/src/main/java/org/openmetadata/service/socket/WebSocketManager.java
@@ -107,7 +107,7 @@ public class WebSocketManager {
             userSocketConnections =
                 activityFeedEndpoints.containsKey(id)
                     ? activityFeedEndpoints.get(id)
-                    : new HashMap<>();
+                    : new ConcurrentHashMap<>();
             userSocketConnections.put(socket.getId(), socket);
             activityFeedEndpoints.put(id, userSocketConnections);
           }