Correct compilation error in unit tests.