regularization fixes
regularization fixes