improve tests // flake8