NThis reads like a typescript limitation more than anything
NYou might have to more explicitly type the Context if inference can’t tell thanks to the branching
NIf you can throw up a reproduction somewhere I can take a look. Fork one of the minimal starters maybe?