Fixed import issue in __init__.