Fix update_nodes function