diff --git a/openmetadata-ui/src/main/resources/ui/playwright/utils/common.ts b/openmetadata-ui/src/main/resources/ui/playwright/utils/common.ts
index a7aafd4af17..266ffb2d3f9 100644
--- a/openmetadata-ui/src/main/resources/ui/playwright/utils/common.ts
+++ b/openmetadata-ui/src/main/resources/ui/playwright/utils/common.ts
@@ -58,6 +58,7 @@ export const getAuthContext = async (token: string) => {
 
 export const redirectToHomePage = async (page: Page) => {
   await page.goto('/');
+  await page.waitForLoadState('networkidle');
   await page.waitForURL('**/my-data');
 };
 
@@ -120,6 +121,10 @@ export const toastNotification = async (
   message: string | RegExp,
   timeout?: number
 ) => {
+  await page.waitForSelector('[data-testid="alert-bar"]', {
+    state: 'visible',
+  });
+
   await expect(page.getByTestId('alert-bar')).toHaveText(message, { timeout });
 
   await expect(page.getByTestId('alert-icon')).toBeVisible();
diff --git a/openmetadata-ui/src/main/resources/ui/playwright/utils/entity.ts b/openmetadata-ui/src/main/resources/ui/playwright/utils/entity.ts
index 8e74ec6962d..68c1b6bb5e8 100644
--- a/openmetadata-ui/src/main/resources/ui/playwright/utils/entity.ts
+++ b/openmetadata-ui/src/main/resources/ui/playwright/utils/entity.ts
@@ -41,6 +41,7 @@ export const visitEntityPage = async (data: {
   const waitForSearchResponse = page.waitForResponse(
     '/api/v1/search/query?q=*index=dataAsset*'
   );
+  await page.waitForLoadState('networkidle');
   await page.getByTestId('searchBox').fill(searchTerm);
   await waitForSearchResponse;
   await page.getByTestId(dataTestId).getByTestId('data-name').click();
@@ -1301,7 +1302,8 @@ export const hardDeleteEntity = async (
   endPoint: EntityTypeEndpoint
 ) => {
   await page.click('[data-testid="manage-button"]');
-  await page.getByTestId('delete-button').click();
+  await page.waitForSelector('[data-testid="delete-button"]');
+  await page.click('[data-testid="delete-button"]');
 
   await page.waitForSelector('[role="dialog"].ant-modal');