Fixed bug with edit page