diff --git a/app/src/main/java/be/ugent/sel/studeez/screens/timer_edit/TimerEditViewModel.kt b/app/src/main/java/be/ugent/sel/studeez/screens/timer_edit/TimerEditViewModel.kt
index 98d060f..3258f24 100644
--- a/app/src/main/java/be/ugent/sel/studeez/screens/timer_edit/TimerEditViewModel.kt
+++ b/app/src/main/java/be/ugent/sel/studeez/screens/timer_edit/TimerEditViewModel.kt
@@ -21,8 +21,9 @@ class TimerEditViewModel @Inject constructor(
         return timerInfo
     }
 
-    fun saveTimer(timerInfo: TimerInfo) {
+    fun saveTimer(timerInfo: TimerInfo, goBack: () -> Unit) {
         timerDAO.updateTimer(timerInfo)
+        goBack()
     }
 
 }
\ No newline at end of file