fix unefficient remove call