diff --git a/app/src/main/java/com/github/libretube/ui/views/CustomExoPlayerView.kt b/app/src/main/java/com/github/libretube/ui/views/CustomExoPlayerView.kt
index 79b5d531e..ef13c0e5a 100644
--- a/app/src/main/java/com/github/libretube/ui/views/CustomExoPlayerView.kt
+++ b/app/src/main/java/com/github/libretube/ui/views/CustomExoPlayerView.kt
@@ -172,7 +172,9 @@ internal class CustomExoPlayerView(
                     )
                 ) {
                     updatePlayPauseButton()
-                    updateScreenOnState()
+
+                    // keep screen on if the video is playing
+                    keepScreenOn = player.isPlaying == true
                 }
             }
         })
@@ -186,11 +188,6 @@ internal class CustomExoPlayerView(
         }
     }
 
-    private fun updateScreenOnState() {
-        // keep screen on if the video is playing
-        keepScreenOn = player?.isPlaying == true
-    }
-
     private fun updatePlayPauseButton() {
         binding.playPauseBTN.setImageResource(
             when {