Good point. I'll extend the tests and correct the code.