Fix header in __init__.py