fix NameError in example