Update __init__.py

South fix.