fixed a return on __init__.py