Fixed remaining pylint error