I'll do this in a future commit - will be pretty simple, but I will have to change the test.