pylint cleanup - found an actual bug this time!