Updating the tests in acorrdance to feedback

https://github.com/rails/rails/pull/9865