Fix style issues

Flake8 is unhappy with current master. I'm not sure why this wasn't breaking earlier.