fix import in the __init__