Fix main.go imports.