Fix the build for unit tests