fix remove node issue