FIX TST due to misuse of api in gbrt