Minor fix to correct error in unit tests.