Fix dodgy unit test.