Fix Pylint warning