Fixed bug in edit page