Fix main.go to build again.