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 ba640e41..e39b3456 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
@@ -278,7 +278,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)
@@ -291,7 +293,7 @@ internal class ExoPlayerPlaybackEngineTest {
     }
 
     @Test
-    fun nextShouldDoNothingIfPlaybackStateIsIdle() = runBlocking {
+    fun nextShouldDoNothingIfPlaybackStateIsIdle() {
         playbackEngine.setNext(forwardingMediaProduct.delegate)
 
         verify(initialExtendedExoPlayer, never())