Fixing pylint error