Update main.go

fix