fix unit test for decision tree