Actually add import to __init__