Fix cmake build

Fix cmake build