Fix missing imports.

These were found with `pylint -E`.