fix example.py