Update __init__.py

Made it play nicely with regular sessions.