Fix compilation error for unit tests.