diff --git a/piface_function.py b/piface_function.py
index aef9260..a4ac03c 100644
--- a/piface_function.py
+++ b/piface_function.py
@@ -199,6 +199,7 @@ class state_machine_day_state(state_machine.state_machine):
             (CONDITION_IDLE, 1, STATE_IDLE),
         ),
         STATE_SUNRISE: (
+            (CONDITION_SLEEP, 1, STATE_SLEEP),
             (CONDITION_SUNSET, 1, STATE_SUNSET),
             (CONDITION_IDLE, 1, STATE_IDLE),
         ),
@@ -208,6 +209,7 @@ class state_machine_day_state(state_machine.state_machine):
         ),
         STATE_SLEEP: (
             (CONDITION_WAKE, 1, STATE_WAKE),
+            (CONDITION_SUNRISE, 1, STATE_SUNRISE),
             (CONDITION_IDLE, 1, STATE_IDLE),
         ),
     }