Fix mock of custom log class.