Minor fix for tests.