fix custom class modal