Revert "Refine imports."

This reverts commit d237e3592ee9d3609d85beee9c043eb7309f0c5f.

pylint reported unused imports, but it was mistaken.