diff --git a/openmetadata-ui/src/main/resources/ui/playwright/e2e/Features/ActivityFeed.spec.ts b/openmetadata-ui/src/main/resources/ui/playwright/e2e/Features/ActivityFeed.spec.ts
index 6fdfe3525a1..71772b35456 100644
--- a/openmetadata-ui/src/main/resources/ui/playwright/e2e/Features/ActivityFeed.spec.ts
+++ b/openmetadata-ui/src/main/resources/ui/playwright/e2e/Features/ActivityFeed.spec.ts
@@ -675,6 +675,7 @@ base.describe('Activity feed with Data Consumer User', () => {
       await commentWithCloseTask;
 
       await toastNotification(page1, 'Task closed successfully.');
+      await page1.waitForLoadState('networkidle');
       await checkTaskCountInActivityFeed(page1, 1, 1);
 
       await afterActionUser1();
@@ -710,6 +711,7 @@ base.describe('Activity feed with Data Consumer User', () => {
       await tagsTask.click();
       await entityPageTaskTab;
 
+      await page2.waitForLoadState('networkidle');
       // Count for task should be 1 both open and closed
 
       checkTaskCountInActivityFeed(page2, 1, 1);