Small change in go.py