Fixed errors for mediator mock