Now correct when there are two nodes