fix expectation of mu-norm-squared