diff --git a/library/core/src/main/java/com/google/android/exoplayer2/WakeLockManager.java b/library/core/src/main/java/com/google/android/exoplayer2/WakeLockManager.java
index f498eea6f4..1e718136e8 100644
--- a/library/core/src/main/java/com/google/android/exoplayer2/WakeLockManager.java
+++ b/library/core/src/main/java/com/google/android/exoplayer2/WakeLockManager.java
@@ -88,11 +88,9 @@ import com.google.android.exoplayer2.util.Log;
   private void updateWakeLock() {
     // Needed for the library nullness check. If enabled is true, the wakelock will not be null.
     if (wakeLock != null) {
-      if (enabled) {
-        if (stayAwake && !wakeLock.isHeld()) {
+      if (enabled && stayAwake) {
+        if (!wakeLock.isHeld()) {
           wakeLock.acquire();
-        } else if (!stayAwake && wakeLock.isHeld()) {
-          wakeLock.release();
         }
       } else if (wakeLock.isHeld()) {
         wakeLock.release();