Revert change in 06fa18d and make sure treemap visualization still works.