Change __init__.py to match type.py