forgot to remove import from __init__.py