That's what I had but flake8 complained. I agree it's better though.