I fixed pylint error introduced in import.