Fixing module name error in __init__