Doh! fix tests.