Fix lint for latest pylint (#155)

2 files changed