Fixes the initial focus when in edit mode.