if  ```n_unique_nodes == 1``` the method is not called. but you are right, I need to change it for the other case.