Update CMakeLists.txt

fix a compile bug in ubuntu