Small fix to schedule controller