diff --git a/package.json b/package.json
index 3eff4130..64652b26 100644
--- a/package.json
+++ b/package.json
@@ -30,7 +30,7 @@
     "test:it:clean": "rimraf ./test-resources && rimraf ./out && rimraf *.vsix"
   },
   "dependencies": {
-    "@kaoto-next/ui": "0.4.0",
+    "@kaoto-next/ui": "0.6.0",
     "@kie-tools-core/backend": "0.32.0",
     "@kie-tools-core/editor": "0.32.0",
     "@kie-tools-core/i18n": "0.32.0",
diff --git a/yarn.lock b/yarn.lock
index fecbf2e6..27b16466 100644
--- a/yarn.lock
+++ b/yarn.lock
@@ -94,9 +94,9 @@ __metadata:
   languageName: node
   linkType: hard
 
-"@kaoto-next/ui@npm:0.4.0":
-  version: 0.4.0
-  resolution: "@kaoto-next/ui@npm:0.4.0"
+"@kaoto-next/ui@npm:0.6.0":
+  version: 0.6.0
+  resolution: "@kaoto-next/ui@npm:0.6.0"
   dependencies:
     "@kaoto-next/uniforms-patternfly": ^0.5.0
     "@kie-tools-core/editor": 0.32.0
@@ -110,6 +110,7 @@ __metadata:
     "@types/uuid": ^9.0.2
     ajv: ^8.12.0
     ajv-formats: ^2.1.1
+    html-to-image: ^1.11.11
     lodash.get: ^4.4.2
     lodash.set: ^4.3.2
     monaco-editor: ^0.34.1
@@ -124,7 +125,7 @@ __metadata:
     uuid: ^9.0.0
     yaml: ^2.3.2
     zustand: ^4.3.9
-  checksum: 2953029922f39f4718f6c93bc8b792a8023c626016da59fecc29dffbb526a085aeeb1cb16f551fc9bfc33f0f63adcfda9d2bfc521142b69be6f6b1fbee1a7cf5
+  checksum: f4272c2f825737a6099a4edd60f867e91516bc7f7d11fc6adab6671e93e9dcabf4cd7df385618c332807f344e1ba9a6fe41c7b7f3bff2956492f3a400bbcb5db
   languageName: node
   linkType: hard
 
@@ -3892,6 +3893,13 @@ __metadata:
   languageName: node
   linkType: hard
 
+"html-to-image@npm:^1.11.11":
+  version: 1.11.11
+  resolution: "html-to-image@npm:1.11.11"
+  checksum: b453beca72a697bf06fae4945e5460d1d9b1751e8569a0d721dda9485df1dde093938cc9bd9172b8df5fc23133a53a4d619777b3d22f7211cd8a67e3197ab4e8
+  languageName: node
+  linkType: hard
+
 "htmlparser2@npm:^8.0.1":
   version: 8.0.2
   resolution: "htmlparser2@npm:8.0.2"
@@ -7602,7 +7610,7 @@ __metadata:
   version: 0.0.0-use.local
   resolution: "vscode-kaoto@workspace:."
   dependencies:
-    "@kaoto-next/ui": 0.4.0
+    "@kaoto-next/ui": 0.6.0
     "@kie-tools-core/backend": 0.32.0
     "@kie-tools-core/editor": 0.32.0
     "@kie-tools-core/i18n": 0.32.0