diff --git a/ui/package.json b/ui/package.json
index 07e566c0a..c22e880c1 100644
--- a/ui/package.json
+++ b/ui/package.json
@@ -20,7 +20,7 @@
     "moment": "^2.29.2",
     "monaco-editor": "^0.44.0",
     "node-forge": "^1.3.0",
-    "orkes-workflow-visualizer": "^1.0.0",
+    "orkes-workflow-visualizer": "^1.0.1",
     "parse-svg-path": "^0.1.2",
     "prop-types": "^15.7.2",
     "react": "^18.3.1",
diff --git a/ui/src/pages/definition/WorkflowDefinition.jsx b/ui/src/pages/definition/WorkflowDefinition.jsx
index 7f18244ca..2fcc3f011 100644
--- a/ui/src/pages/definition/WorkflowDefinition.jsx
+++ b/ui/src/pages/definition/WorkflowDefinition.jsx
@@ -378,6 +378,7 @@ export default function Workflow() {
           {dag && dag?.workflowDef && (
             <WorkflowVisualizer
               maxHeightOverride
+              maxWidthOverride
               pannable
               zoomable
               zoom={0.7}
diff --git a/ui/src/pages/execution/TaskDetails.jsx b/ui/src/pages/execution/TaskDetails.jsx
index 022894d8e..c62b6fbe3 100644
--- a/ui/src/pages/execution/TaskDetails.jsx
+++ b/ui/src/pages/execution/TaskDetails.jsx
@@ -39,6 +39,7 @@ export default function TaskDetails({
         {tabIndex === 0 && (
           <WorkflowVisualizer
             maxHeightOverride
+            maxWidthOverride
             pannable
             zoomable
             zoom={0.7}
@@ -48,9 +49,9 @@ export default function TaskDetails({
               const selectedTaskRefName =
                 data?.data?.task?.executionData?.status === "PENDING"
                   ? pendingTaskSelection(data?.data?.task)?.workflowTask
-                    ?.taskReferenceName
+                      ?.taskReferenceName
                   : taskWithLatestIteration(execution?.tasks, { ref: data.id })
-                    ?.referenceTaskName;
+                      ?.referenceTaskName;
               setSelectedNode(data);
               setSelectedTask({ ref: selectedTaskRefName });
             }}
diff --git a/ui/yarn.lock b/ui/yarn.lock
index 6b4ad3efb..16e7fb2d3 100644
--- a/ui/yarn.lock
+++ b/ui/yarn.lock
@@ -9046,10 +9046,10 @@ optionator@^0.9.1:
     type-check "^0.4.0"
     word-wrap "^1.2.5"
 
-orkes-workflow-visualizer@^1.0.0:
-  version "1.0.0"
-  resolved "https://registry.npmjs.org/orkes-workflow-visualizer/-/orkes-workflow-visualizer-1.0.0.tgz"
-  integrity sha512-74AP8HZeAnWja8oQCgkexik9P+oBt9EBdM13joip1VdUOklIDknSFCRKDmtNXERxU/STEcU9hEDlwOvdl3Za4w==
+orkes-workflow-visualizer@^1.0.1:
+  version "1.0.1"
+  resolved "https://registry.yarnpkg.com/orkes-workflow-visualizer/-/orkes-workflow-visualizer-1.0.1.tgz#18b48a4ab00673cdfd3ee070bdc1083d4df296ea"
+  integrity sha512-vIoliMzp3q6CB0z8kIFYgbqF/VtOINgbb9QvmX8ALWXdIHEQoCKL7j+w9Y1fns0WoWiIny/TVqxkeVPdUjXbMg==
   dependencies:
     date-fns "^2.29.3"
     lodash "^4.17.21"