fixing import in __init__.py after removing aliases