resolve issue with class breaking

TypeError

wrong argument type Class (expected Module)