Small interface modification on TB.cpp