Fix go vet errors