FeatureState doesn't play nicely with hot reloading

[livtanong]

Describe the bug
When I hot reload I get the following error message:
Error: The source '{{source-id}}' does not exist in the map's style.
I'm guessing this is because no cleanup for feature states happens during unmounting of the Source component

To Reproduce
Just use feature state, then trigger a hot reload

Expected behavior
No error

Desktop (please complete the following information):

• OS: Windows 10
• Browser: Microsoft Edge (Chromium)
• Version: 88.0.705.81

Additional context
I'm aware the solution cannot be the same as with Layer cleanup shown here, but I think it's possible to use queryRenderedFeatures in conjunction with removeFeatureState.