fix error on remove