Flake8 fixes - eesh