Minor update to CMake function.