diff --git a/packages/ui/src/ui-component/dialog/FormatPromptValuesDialog.js b/packages/ui/src/ui-component/dialog/FormatPromptValuesDialog.js
index 233f0762..6afb84cc 100644
--- a/packages/ui/src/ui-component/dialog/FormatPromptValuesDialog.js
+++ b/packages/ui/src/ui-component/dialog/FormatPromptValuesDialog.js
@@ -28,7 +28,7 @@ const FormatPromptValuesDialog = ({ show, dialogProps, onChange, onCancel }) =>
             aria-describedby='alert-dialog-description'
         >
             <DialogTitle sx={{ fontSize: '1rem' }} id='alert-dialog-title'>
-                Format Prompt Values
+                {dialogProps.inputParam.label ?? 'Format Prompt Values'}
             </DialogTitle>
             <DialogContent>
                 <PerfectScrollbar