PEP 8 fix/logic error
[skip ci]