Correction after pylint