trying to fix treemap(for now)