fix: focus edit mode