I prefer to add the test in this PR.

It's a good rule: if you touch a piece of code, you make it better.