fix build in Demo