Minor fix to tests.