diff --git a/app/src/main/java/be/ugent/sel/studeez/screens/session/sessionScreens/composables/GetSessionScreenComposable.kt b/app/src/main/java/be/ugent/sel/studeez/screens/session/sessionScreens/composables/GetSessionScreenComposable.kt
index bae4d92..3780f42 100644
--- a/app/src/main/java/be/ugent/sel/studeez/screens/session/sessionScreens/composables/GetSessionScreenComposable.kt
+++ b/app/src/main/java/be/ugent/sel/studeez/screens/session/sessionScreens/composables/GetSessionScreenComposable.kt
@@ -1,19 +1,15 @@
 package be.ugent.sel.studeez.screens.session.sessionScreens.composables
 
-import android.media.MediaPlayer
 import androidx.compose.runtime.Composable
 import be.ugent.sel.studeez.data.local.models.timer_functional.FunctionalCustomTimer
 import be.ugent.sel.studeez.data.local.models.timer_functional.FunctionalEndlessTimer
 import be.ugent.sel.studeez.data.local.models.timer_functional.FunctionalPomodoroTimer
 import be.ugent.sel.studeez.data.local.models.timer_functional.FunctionalTimerVisitor
 import be.ugent.sel.studeez.screens.session.SessionActions
-import be.ugent.sel.studeez.screens.session.sessionScreens.AbstractSessionScreen
-import be.ugent.sel.studeez.screens.session.sessionScreens.BreakSessionScreen
-import be.ugent.sel.studeez.screens.session.sessionScreens.CustomSessionScreen
-import be.ugent.sel.studeez.screens.session.sessionScreens.EndlessSessionScreen
+import be.ugent.sel.studeez.screens.session.SoundPlayer
 
 class GetSessionScreenComposable(
-    private val mediaplayer: MediaPlayer?,
+    private val soundPlayer: SoundPlayer,
     private val open: (String) -> Unit,
     private val sessionActions: SessionActions
     ) :
@@ -23,7 +19,8 @@ class GetSessionScreenComposable(
         return { CustomTimerSessionScreenComposable(
                 open = open,
                 sessionActions = sessionActions,
-                customTimer = functionalCustomTimer
+                soundPlayer = soundPlayer,
+                customTimer = functionalCustomTimer,
             )
         }
     }
@@ -42,6 +39,7 @@ class GetSessionScreenComposable(
             BreakSessionScreenComposable(
                 open = open,
                 sessionActions = sessionActions,
+                soundPlayer = soundPlayer,
                 pomodoroTimer = functionalPomodoroTimer
             )
         }