fixing a bug in the decision tree model