Fix busy loop (sleep wouldn't be triggered when the display is off)

afterlock/afterlock.go

@@ -112,15 +112,15 @@ func (af *AfterLock) hybernateDisplayLoop() {
 			if err != nil {
 				panic(err)
 			}
-			for {
-				time.Sleep(time.Duration(af.LoopDelay) * time.Second)
-				if af.keypressFlag.Get() {
-					af.keypressFlag.Set(false)
-					continue
-				} else {
-					break
-				}
+		}
+
+		for {
+			time.Sleep(time.Duration(af.LoopDelay) * time.Second)
+			if af.keypressFlag.Get() {
+				af.keypressFlag.Set(false)
+				continue
 			}
+			break
 		}
 	}
 }