fix bug with example.py