diff --git a/docs/docs/components/custom.mdx b/docs/docs/components/custom.mdx
index cbc5f13ff..10b45c9b7 100644
--- a/docs/docs/components/custom.mdx
+++ b/docs/docs/components/custom.mdx
@@ -27,7 +27,7 @@ A custom component in Python looks like this:
 
 ```python
 from langflow.custom import Component
-from langflow.inputs import TextInput, IntInput, BoolInput, DropdownInput, HandleInput
+from langflow.inputs import MessageTextInput, IntInput, BoolInput, DropdownInput, HandleInput
 from langflow.template import Output
 from langflow.schema import Data, Message
 from typing import List, Optional