Fixing pylint