Update __init__.py

Fix the mess of the previous merge.