Tidies up fix for mediator reset