diff --git a/services/core/java/com/android/server/power/PowerManagerService.java b/services/core/java/com/android/server/power/PowerManagerService.java
index c4aca6c184536..03722cd9afb4e 100644
--- a/services/core/java/com/android/server/power/PowerManagerService.java
+++ b/services/core/java/com/android/server/power/PowerManagerService.java
@@ -2956,8 +2956,7 @@ public final class PowerManagerService extends SystemService
                 // Group has been removed.
                 return;
             }
-            // TODO (b/175764708): Support per-display doze.
-            wakefulness = getWakefulnessLocked();
+            wakefulness = mDisplayGroupPowerStateMapper.getWakefulnessLocked(groupId);
             if ((wakefulness == WAKEFULNESS_DREAMING || wakefulness == WAKEFULNESS_DOZING) &&
                     mDisplayGroupPowerStateMapper.isSandmanSummoned(groupId)
                     && mDisplayGroupPowerStateMapper.isReady(groupId)) {
@@ -3099,6 +3098,7 @@ public final class PowerManagerService extends SystemService
      * Returns true if the device is allowed to doze in its current state.
      */
     private boolean canDozeLocked() {
+        // TODO (b/175764708): Support per-display doze.
         return getWakefulnessLocked() == WAKEFULNESS_DOZING;
     }