Minor fix to test.