Minor fix for the tests.