Fix incorrect change made to HuberLoss