fix remove functionality