Fix edit/cancel bug