Little fix
Fix edit action bug