fix pylint

I always forget these