Fix a small issue found with go vet.
Forked (rewrite imports).