Fix gofmt issues