Minor Flake8 fix.

Fixes #1208. [skip ci]