Minor fix to add @class before @protocol