Change __init__.py to reflect move of Context class.