fix treebank controllerS