Fix flake8 error on __init__.py