Fix test around removing