fixed a bug on example.py