diff --git a/tests/testutils/src/main/kotlin/io/element/android/tests/testutils/WithFakeLifecycleOwner.kt b/tests/testutils/src/main/kotlin/io/element/android/tests/testutils/WithFakeLifecycleOwner.kt
index 16d4cfb2fd..20bc431033 100644
--- a/tests/testutils/src/main/kotlin/io/element/android/tests/testutils/WithFakeLifecycleOwner.kt
+++ b/tests/testutils/src/main/kotlin/io/element/android/tests/testutils/WithFakeLifecycleOwner.kt
@@ -12,8 +12,6 @@ import androidx.compose.runtime.Composable
 import androidx.compose.runtime.InternalComposeApi
 import androidx.compose.runtime.Stable
 import androidx.compose.runtime.currentComposer
-import androidx.compose.runtime.getValue
-import androidx.compose.runtime.setValue
 import androidx.lifecycle.Lifecycle
 import androidx.lifecycle.LifecycleOwner
 import androidx.lifecycle.LifecycleRegistry
@@ -26,8 +24,6 @@ import io.element.android.libraries.architecture.Presenter
 
 /**
  * Composable that provides a fake [LifecycleOwner] to the composition.
- *
- * **WARNING: DO NOT USE OUTSIDE TESTS.**
  */
 @OptIn(InternalComposeApi::class)
 @Stable
@@ -44,19 +40,16 @@ fun <T> withFakeLifecycleOwner(
 
 /**
  * Test a [Presenter] with a fake [LifecycleOwner].
- *
- * **WARNING: DO NOT USE OUTSIDE TESTS.**
  */
 suspend fun <T> Presenter<T>.testWithLifecycleOwner(
     lifecycleOwner: FakeLifecycleOwner = FakeLifecycleOwner(),
     block: suspend TurbineTestContext<T>.() -> Unit
 ) {
     moleculeFlow(RecompositionMode.Immediate) {
-        val ret = withFakeLifecycleOwner(lifecycleOwner) {
+        withFakeLifecycleOwner(lifecycleOwner) {
             present()
         }
-        ret
-    }.test<T>(validate = block)
+    }.test(validate = block)
 }
 
 @SuppressLint("VisibleForTests")