fix tst bug