Fixed 'go vet' issue