diff --git a/app/src/androidTest/java/app/hermes/mobile/feature/chat/ChatViewModelScopeTest.kt b/app/src/androidTest/java/app/hermes/mobile/feature/chat/ChatViewModelScopeTest.kt
index 6ff09c0..1756800 100644
--- a/app/src/androidTest/java/app/hermes/mobile/feature/chat/ChatViewModelScopeTest.kt
+++ b/app/src/androidTest/java/app/hermes/mobile/feature/chat/ChatViewModelScopeTest.kt
@@ -1,6 +1,6 @@
 package app.hermes.mobile.feature.chat
 
-import androidx.compose.ui.test.assertExists
+import androidx.compose.ui.test.assertIsDisplayed
 import androidx.compose.ui.test.junit4.createAndroidComposeRule
 import androidx.compose.ui.test.onAllNodesWithText
 import androidx.compose.ui.test.onNodeWithText
@@ -52,6 +52,6 @@ class ChatViewModelScopeTest {
             composeTestRule.onAllNodesWithText("Draft prompt before rotation", substring = true)
                 .fetchSemanticsNodes().isNotEmpty()
         }
-        composeTestRule.onNodeWithText("Draft prompt before rotation", substring = true).assertExists()
+        composeTestRule.onNodeWithText("Draft prompt before rotation", substring = true).assertIsDisplayed()
     }
 }