STY: Tweak over-indented code.

flake8 recommendation