Minor fix to a test.