Yep, makes sense to me. I've updated the tests.