Fix old imports in __init__.py