Fix cmake file for unit test