Fix for CMakeLists.txt