Small Fix with edit page