Fixing imports for go get.