Fix PyLint warning