Fix from go vet