diff --git a/docs/docs/modules/model_io/models/chat/index.mdx b/docs/docs/modules/model_io/models/chat/index.mdx
index 90d219bb38ac..a8fe816ec792 100644
--- a/docs/docs/modules/model_io/models/chat/index.mdx
+++ b/docs/docs/modules/model_io/models/chat/index.mdx
@@ -1,6 +1,7 @@
 ---
 sidebar_position: 1
 ---
+import CodeBlock from "@theme/CodeBlock";
 
 # Chat models