fixed wrong MSE output