fix test by adding mu and sigma to expected