Minor fix to one of the tests.