fix for regression tree