Make example work with latest changes