Fix testRemove method
Fix testRemove method