Fix tests in CMake build