Pylint cleanup

Fix my pylint :)