Add tests for working before bedtime