fix to regularization term