flake8 style fixes for #212

Fixes #212