Minor changes for pylint