diff --git a/app/src/androidTest/java/org/kiwix/kiwixmobile/main/TopLevelDestinationTest.kt b/app/src/androidTest/java/org/kiwix/kiwixmobile/main/TopLevelDestinationTest.kt
index 9b60a862c..0471cd038 100644
--- a/app/src/androidTest/java/org/kiwix/kiwixmobile/main/TopLevelDestinationTest.kt
+++ b/app/src/androidTest/java/org/kiwix/kiwixmobile/main/TopLevelDestinationTest.kt
@@ -119,13 +119,13 @@ class TopLevelDestinationTest : BaseActivityTest() {
         }
       }
       clickBookmarksOnNavDrawer {
-        assertBookMarksDisplayed()
-        clickOnTrashIcon()
+        assertBookMarksDisplayed(composeTestRule)
+        clickOnTrashIcon(composeTestRule)
         assertDeleteBookmarksDialogDisplayed()
       }
       clickHistoryOnSideNav {
-        assertHistoryDisplayed()
-        clickOnTrashIcon()
+        assertHistoryDisplayed(composeTestRule)
+        clickOnTrashIcon(composeTestRule)
         assertDeleteHistoryDialogDisplayed()
       }
       clickHostBooksOnSideNav(ZimHostRobot::assertMenuWifiHotspotDiplayed)