Small bug fix - treetable needed but doesn't exist