Fixing plotting bug for treeVars