Fixing pylint violation