Fix demo build