diff --git a/libraries/session-storage/test/src/main/kotlin/io/element/android/libraries/sessionstorage/test/InMemorySessionStore.kt b/libraries/session-storage/test/src/main/kotlin/io/element/android/libraries/sessionstorage/test/InMemorySessionStore.kt
index 95968afb8f..8228adada5 100644
--- a/libraries/session-storage/test/src/main/kotlin/io/element/android/libraries/sessionstorage/test/InMemorySessionStore.kt
+++ b/libraries/session-storage/test/src/main/kotlin/io/element/android/libraries/sessionstorage/test/InMemorySessionStore.kt
@@ -54,7 +54,7 @@ class InMemorySessionStore(
     }
 
     override suspend fun getSession(sessionId: String): SessionData? {
-        return sessionDataListFlow.value.first { it.userId == sessionId }
+        return sessionDataListFlow.value.firstOrNull { it.userId == sessionId }
     }
 
     override suspend fun getAllSessions(): List<SessionData> {