Fix bug in regularization