Fix pylint2 issue