Fix pylint warning