Change focus to search box

This one hits me dozens of times a day.

I go to insert a new element,
change from Insert Child to Insert Parent
and then I start typing the name of an element,

but unfortunately, after changing to Insert Parent, the focus is no longer on the search box. It would be very helpful to reset focus to the search box after any change to the category or insert position.