Minor updates for pylint check