fixing import for __init__.py