minor bug fix in example.py