fix remove action behavior