diff --git a/app/client/src/components/ads/TreeDropdown.tsx b/app/client/src/components/ads/TreeDropdown.tsx
index f4fbb5ade7..099fb97b61 100644
--- a/app/client/src/components/ads/TreeDropdown.tsx
+++ b/app/client/src/components/ads/TreeDropdown.tsx
@@ -13,7 +13,7 @@ import {
 import styled from "styled-components";
 import { Colors } from "constants/Colors";
 import { DropdownOption } from "components/constants";
-import Icon, { IconName, IconSize } from "components/ads/Icon";
+import Icon, { IconSize } from "components/ads/Icon";
 
 export type TreeDropdownOption = DropdownOption & {
   onSelect?: (value: TreeDropdownOption, setter?: Setter) => void;
@@ -144,10 +144,6 @@ function getSelectedOption(
   return selectedOption;
 }
 
-const getMoreIcons = (icon: React.ReactNode) => {
-  return <Icon name={icon as IconName} size={IconSize.XXL} />;
-};
-
 export default function TreeDropdown(props: TreeDropdownProps) {
   const {
     defaultText,
@@ -184,7 +180,7 @@ export default function TreeDropdown(props: TreeDropdownProps) {
       <MenuItem
         active={isSelected}
         className={option.className || "single-select"}
-        icon={getMoreIcons(option.icon)}
+        icon={option.icon}
         intent={option.intent}
         key={option.value}
         onClick={
@@ -218,7 +214,7 @@ export default function TreeDropdown(props: TreeDropdownProps) {
         className={`t--open-dropdown-${defaultText.split(" ").join("-")} ${
           selectedLabelModifier ? "code-highlight" : ""
         }`}
-        rightIcon={getMoreIcons("downArrow")}
+        rightIcon={<Icon name="downArrow" size={IconSize.XXL} />}
         text={
           selectedLabelModifier
             ? selectedLabelModifier(selectedOption, displayValue)