Fix to work with go get