chore: changed demo to fit latest lib changes