Update __init__.py

Fixed Vincent's comment.