fix for SetClass