Refactor to use module augmentation instead of @ts-expect-error

[remcohaszing]

Initial checklist

• I read the support docs
• I read the contributing guide
• I agree to follow the code of conduct
• I searched issues and couldn’t find anything (or linked relevant results below)
• If applicable, I’ve added docs and tests

Description of changes

This removes the need for some @ts-expect-error comments.

[wooorm]

I don’t think that “mocking” the react types is better than a @ts-expect-error but fine if you want this