Minor update to CMake file