fix the build
Fixed some go vet warnings