cleanup for pylint