Fix demo after library changes