Fix last pylint violation.

Nice Fix Akbar! Perfect Idea!