Why not do this in the assign-heights traversal?