flake8 cleanup in test