Add missing import.
Pylint fixes.