Correct error in unit tests