Fix edit action bug