scheduler: Wrap in leader wait