fix bug in remove