I think `return node_ids[0]` would be a bit clearer here.