Fixed pylint C0326 issue