Oh. thats a good catch. I'll add the check. Also, I am working on another PR that adds `reset` in current scheduler