Fix on remove_node