Fix the edit page - create and edit elements work