Ensure to generate input_class only for the input, not for the wrapper

Thanks @ollym for catching this one. Related to #337.