Fix focus on edit