Fix main.go.