Fix import statement in __init__