fix test of remove method