diff --git a/docling_core/types/doc/document.py b/docling_core/types/doc/document.py
index 744059c..9b99d38 100644
--- a/docling_core/types/doc/document.py
+++ b/docling_core/types/doc/document.py
@@ -1313,6 +1313,7 @@ class KeyValueItem(DocItem):
         PictureItem,
         TableItem,
         KeyValueItem,
+        CodeItem,
     ],
     Field(discriminator="label"),
 ]