Fix bug in tree column