Update __init__.py

miss return the value.