fix build for demo