Minor fix to unit tests.