Fix mse bug