Minor fix on tests.