Fix the example wrt. recent changes.