diff --git a/features/messages/impl/src/test/kotlin/io/element/android/features/messages/MessagesPresenterTest.kt b/features/messages/impl/src/test/kotlin/io/element/android/features/messages/MessagesPresenterTest.kt
index aeca219b68..a24e5f9ff8 100644
--- a/features/messages/impl/src/test/kotlin/io/element/android/features/messages/MessagesPresenterTest.kt
+++ b/features/messages/impl/src/test/kotlin/io/element/android/features/messages/MessagesPresenterTest.kt
@@ -490,11 +490,11 @@ class MessagesPresenterTest {
         }.test {
             val initialState = consumeItemsUntilTimeout().last()
             initialState.eventSink(MessagesEvents.InviteDialogDismissed(InviteDialogAction.Invite))
-            skipItems(1)
-            val loadingState = awaitItem()
-            assertThat(loadingState.inviteProgress.isLoading()).isTrue()
-            val newState = awaitItem()
-            assertThat(newState.inviteProgress.isFailure()).isTrue()
+            val remainingStates = consumeItemsUntilTimeout()
+            assertThat(remainingStates.size).isEqualTo(3)
+            assertThat(remainingStates[0].inviteProgress.isLoading()).isFalse()
+            assertThat(remainingStates[1].inviteProgress.isLoading()).isTrue()
+            assertThat(remainingStates[2].inviteProgress.isFailure()).isTrue()
         }
     }