Do you mean remove an *implied* speciation node?  The assert for `len(node.children) == 1` makes me think this is what you intend.

Given this interpretation, function looks good.