fix for Treevisitor Classnot found error