diff --git a/public/pages/workflow_detail/workflow_inputs/export_options/export_options.tsx b/public/pages/workflow_detail/workflow_inputs/export_options/export_options.tsx
index dc6eebce..409ca35e 100644
--- a/public/pages/workflow_detail/workflow_inputs/export_options/export_options.tsx
+++ b/public/pages/workflow_detail/workflow_inputs/export_options/export_options.tsx
@@ -70,7 +70,7 @@ export function ExportOptions(props: SearchInputsProps) {
       {props.workflow !== undefined && (
         <EuiFlexItem grow={false}>
           <EuiCodeBlock
-            language={selectedOption === EXPORT_OPTION.JSON ? 'json' : 'yaml'}
+            language={selectedOption}
             fontSize="m"
             isCopyable={true}
           >