minor fix on remove method