Minor update for go code