Fix PyLint errors