[*] fix import in __init__