I don't see the fix for the regression case, only on the previous classification test.