We could fix this by providing an onRest callback which, when called, will allow the user to properly focus in the active pane using the imperative api.
Alternatively, perhaps we can handle autoFocus inside the component since most devs might be too lazy to actually implement this properly anyway. In this case, we would only autofocus when it changes (not on mount)