Minor fix for a failing test.