Fixed the `testRemove` method