fixed import in __init__