Fix error in edit layout