Fixed remove node bug