Fix layout in add and edit register