fixed issue with labels in final tree