Bah, ho risolto con un for loop. Incredibile quante volte penso a walk e tree-seq quando invece dovrei usare un for. Lesson learned