Small change to RMSE test