Fix name errors in __init__.py
Attempt to fix module error in __init__.py