Fix __init__ file