Make flake8 happy for tests