Put __init__ code in the right __init__.py.