Update CMakeLists.txt

Fixes the build.