Small error dealing with remove node