diff --git a/components/lib/treeselect/TreeSelect.js b/components/lib/treeselect/TreeSelect.js
index 43b0ec1963..c2cc9a4e6f 100644
--- a/components/lib/treeselect/TreeSelect.js
+++ b/components/lib/treeselect/TreeSelect.js
@@ -797,7 +797,12 @@ export const TreeSelect = React.memo(
                     props
                 };
 
-                return ObjectUtils.getJSXElement(props.panelHeaderTemplate, defaultOptions);
+                return (
+                    <div>
+                        {content}
+                        {ObjectUtils.getJSXElement(props.panelHeaderTemplate, defaultOptions)}
+                    </div>
+                );
             }
 
             return content;