Fixed __init__.py import