diff --git a/player/playback-engine/src/test/kotlin/com/tidal/sdk/player/playbackengine/ExoPlayerPlaybackEngineTest.kt b/player/playback-engine/src/test/kotlin/com/tidal/sdk/player/playbackengine/ExoPlayerPlaybackEngineTest.kt
index 17c935ed..874a369e 100644
--- a/player/playback-engine/src/test/kotlin/com/tidal/sdk/player/playbackengine/ExoPlayerPlaybackEngineTest.kt
+++ b/player/playback-engine/src/test/kotlin/com/tidal/sdk/player/playbackengine/ExoPlayerPlaybackEngineTest.kt
@@ -274,7 +274,9 @@ internal class ExoPlayerPlaybackEngineTest {
     }
 
     @Test
-    fun loadShouldSetCorrectPropertiesAndPreparePlayerWithCorrectMediaSource() = runBlocking {
+    fun loadShouldSetCorrectPropertiesAndPreparePlayerWithCorrectMediaSource() {
+        whenever(initialExtendedExoPlayer.load(any())) doReturn mediaSource
+
         playbackEngine.load(forwardingMediaProduct.delegate)
 
         assertThat(playbackEngine.mediaProduct).isEqualTo(forwardingMediaProduct.delegate)
@@ -287,7 +289,7 @@ internal class ExoPlayerPlaybackEngineTest {
     }
 
     @Test
-    fun nextShouldDoNothingIfPlaybackStateIsIdle() = runBlocking {
+    fun nextShouldDoNothingIfPlaybackStateIsIdle() {
         playbackEngine.setNext(forwardingMediaProduct.delegate)
 
         verify(initialExtendedExoPlayer, never())