Fix main.go