Fixed __init__ import