Fix two small flake8 violations