Add the missing CustomAuditRepository class