Fixing import for go get.