diff --git a/afterlock/afterlock.go b/afterlock/afterlock.go
index 74360aa..001cace 100644
--- a/afterlock/afterlock.go
+++ b/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
 		}
 	}
 }
\ No newline at end of file