Fixed test case to adjust for CBForest.