small change on remove