Fixed little bug in remove.