diff --git a/app/client/src/pages/DataGenerator/Prompt.tsx b/app/client/src/pages/DataGenerator/Prompt.tsx
index 4638806..e27788d 100644
--- a/app/client/src/pages/DataGenerator/Prompt.tsx
+++ b/app/client/src/pages/DataGenerator/Prompt.tsx
@@ -19,6 +19,7 @@ import TextArea from 'antd/es/input/TextArea';
 import FileSelectorButton from './FileSelectorButton';
 import { useMutation } from '@tanstack/react-query';
 import first from 'lodash/first';
+import ResetIcon from './ResetIcon';
 
 const { Title } = Typography;
 
@@ -254,6 +255,33 @@ const Prompt = () => {
                                             setPrompt={setPrompt}
                                         />
                                     }
+                                    <RestoreDefaultBtn
+                                        icon={<ResetIcon />}
+                                        type="link"
+                                        onClick={() => {
+                                            return Modal.warning({
+                                                title: 'Restore Default Prompt',
+                                                closable: true,
+                                                content: <>{'Are you sure you want to restore to the default prompt? Your current prompt will be lost.'}</>,
+                                                footer: (
+                                                    <ModalButtonGroup gap={8} justify='end'>
+                                                        <Button onClick={() => Modal.destroyAll()}>{'Cancel'}</Button>
+                                                        <Button
+                                                            onClick={() => {
+                                                                form.setFieldValue('custom_prompt', defaultPromptRef.current)
+                                                                Modal.destroyAll()
+                                                            }}
+                                                            type='primary'
+                                                        >
+                                                            {'Confirm'}
+                                                        </Button>
+                                                    </ModalButtonGroup>
+                                                ),
+                                                maskClosable: true,
+                                            })
+                                        }}>
+                                            {'Restore'}
+                                        </RestoreDefaultBtn>    
                                     </Flex>
                                 </div>
                             }
@@ -263,31 +291,7 @@ const Prompt = () => {
                         >
                             <StyledTextArea autoSize placeholder={'Enter a prompt'}/>
                         </StyledPromptFormItem>
-                        <RestoreDefaultBtn
-                            onClick={() => {
-                                return Modal.warning({
-                                    title: 'Restore Default Prompt',
-                                    closable: true,
-                                    content: <>{'Are you sure you want to restore to the default prompt? Your current prompt will be lost.'}</>,
-                                    footer: (
-                                        <ModalButtonGroup gap={8} justify='end'>
-                                            <Button onClick={() => Modal.destroyAll()}>{'Cancel'}</Button>
-                                            <Button
-                                                onClick={() => {
-                                                    form.setFieldValue('custom_prompt', defaultPromptRef.current)
-                                                    Modal.destroyAll()
-                                                }}
-                                                type='primary'
-                                            >
-                                                {'Confirm'}
-                                            </Button>
-                                        </ModalButtonGroup>
-                                    ),
-                                    maskClosable: true,
-                                })
-                            }}>
-                                {'Restore'}
-                        </RestoreDefaultBtn>
+                        
                         
                     </div>
                     {((workflow_type === WorkflowType.CUSTOM_DATA_GENERATION && !isEmpty(doc_paths)) ||
diff --git a/app/client/src/pages/DataGenerator/ResetIcon.tsx b/app/client/src/pages/DataGenerator/ResetIcon.tsx
new file mode 100644
index 0000000..053384b
--- /dev/null
+++ b/app/client/src/pages/DataGenerator/ResetIcon.tsx
@@ -0,0 +1,21 @@
+import React from 'react';
+
+// eslint-disable-next-line @typescript-eslint/no-explicit-any
+const ResetIcon = (props: any) =>
+  /*#__PURE__*/ React.createElement(
+    "svg",
+    {
+      viewBox: "0 0 24 24",
+      fill: "none",
+      width: "1em",
+      height: "1em",
+      className: "cdp-icon-ai-assistant",
+      ...props,
+    },
+    /*#__PURE__*/ React.createElement("path", {
+      d: "M11.013 10.99h-7.99V3h1.998v3.39c1.646-2.048 4.163-3.367 6.991-3.367 4.956 0 8.988 4.032 8.988 8.988 0 4.957-4.032 8.99-8.988 8.99-3.905 0-7.227-2.508-8.464-5.993h2.157c1.124 2.357 3.525 3.995 6.307 3.995 3.855 0 6.99-3.137 6.99-6.992 0-3.854-3.135-6.99-6.99-6.99-2.772 0-5.164 1.626-6.294 3.971h5.295v1.997z",
+      fill: "currentColor",
+    }),
+  );
+
+export default ResetIcon;
\ No newline at end of file
diff --git a/app/client/src/pages/Evaluator/FreeFromEvaluationTable.tsx b/app/client/src/pages/Evaluator/FreeFromEvaluationTable.tsx
index 6e905a8..c443921 100644
--- a/app/client/src/pages/Evaluator/FreeFromEvaluationTable.tsx
+++ b/app/client/src/pages/Evaluator/FreeFromEvaluationTable.tsx
@@ -18,7 +18,7 @@ import {
  import { themeMaterial } from "ag-grid-community";
 import get from 'lodash/get';
 import { getColorCode } from './util';
-import { Badge, Popover, Tooltip } from 'antd';
+import { Badge, Popover } from 'antd';
 import styled from 'styled-components';
 
 interface Props {