edit bug.py

edit bug.py