Ajax replace inside a modal causes the current field to lose focus

Ajax replace in a dialog, such as a tabledrag widget for an entity reference field, will reload the form and refocus on the first element.