[UnitTests] Fix build.