fixes small type in __init__.py