Replaced the component to also work for schedules