From 0065b215a944549112fe9bae6823a8d0f1357429 Mon Sep 17 00:00:00 2001
From: Trenton Holmes <797416+stumpylog@users.noreply.github.com>
Date: Fri, 3 Nov 2023 06:59:06 -0700
Subject: [PATCH] Register the new models with auditlog

---
 src/documents/models.py | 18 ++++++++++--------
 1 file changed, 10 insertions(+), 8 deletions(-)

diff --git a/src/documents/models.py b/src/documents/models.py
index d1ced4359..233faac2e 100644
--- a/src/documents/models.py
+++ b/src/documents/models.py
@@ -877,14 +877,6 @@ class ConsumptionTemplate(models.Model):
         return f"{self.name}"
 
 
-if settings.AUDIT_LOG_ENABLED:
-    auditlog.register(Document, m2m_fields={"tags"})
-    auditlog.register(Correspondent)
-    auditlog.register(Tag)
-    auditlog.register(DocumentType)
-    auditlog.register(Note)
-
-
 class CustomField(models.Model):
     """
     Defines the name and type of a custom field
@@ -1001,3 +993,13 @@ class CustomFieldInstance(models.Model):
         elif self.field.data_type == CustomField.FieldDataType.INT:
             return self.value_int
         raise NotImplementedError(self.field.data_type)
+
+
+if settings.AUDIT_LOG_ENABLED:
+    auditlog.register(Document, m2m_fields={"tags"})
+    auditlog.register(Correspondent)
+    auditlog.register(Tag)
+    auditlog.register(DocumentType)
+    auditlog.register(Note)
+    auditlog.register(CustomField)
+    auditlog.register(CustomFieldInstance)