fix bug in example.py