fix protocol class