Fixed BUG with MSE