diff --git a/app/src/test/java/org/justinvisser/wolfskin/tv/LibraryProgressiveSurfaceTest.kt b/app/src/test/java/org/justinvisser/wolfskin/tv/LibraryProgressiveSurfaceTest.kt
index e312a66d..976f9588 100644
--- a/app/src/test/java/org/justinvisser/wolfskin/tv/LibraryProgressiveSurfaceTest.kt
+++ b/app/src/test/java/org/justinvisser/wolfskin/tv/LibraryProgressiveSurfaceTest.kt
@@ -74,8 +74,10 @@ class LibraryProgressiveSurfaceTest {
                 org.robolectric.Shadows.shadowOf(android.os.Looper.getMainLooper()).idle()
                 shell.view.library?.source?.size == 201
             }
-            assertEquals(first[1].id, shell.view.library!!.focusedTitle!!.id)
-            assertFalse(shell.navFocused)
+            rule.runOnIdle {
+                assertEquals(first[1].id, shell.view.library!!.focusedTitle!!.id)
+                assertFalse(shell.navFocused)
+            }
         } finally { release.complete(Unit) }
     }
 }