diff --git a/packages/ui/core/src/assets/img/custom/piece/code.svg b/packages/ui/core/src/assets/img/custom/piece/code.svg
index 0355f21bed..fa83feb2af 100644
--- a/packages/ui/core/src/assets/img/custom/piece/code.svg
+++ b/packages/ui/core/src/assets/img/custom/piece/code.svg
@@ -1,5 +1,11 @@
-<svg width="49" height="49" viewBox="0 0 49 49" fill="none" xmlns="http://www.w3.org/2000/svg">
-<rect x="0.499939" y="0.337891" width="48" height="48" rx="5" fill="#D9A032" fill-opacity="0.1"/>
-<path d="M16.2932 20.2219V17.8165C16.2932 15.158 18.3467 13 20.8794 13V15.409C19.6139 15.409 18.5861 16.4874 18.5861 17.8165V20.2219C18.5861 21.6687 17.966 22.9519 17.0039 23.835C17.966 24.7177 18.5861 25.9991 18.5861 27.4467V29.8542C18.5861 31.184 19.6139 32.2621 20.8804 32.2621V34.67C18.3491 34.67 16.2932 32.5149 16.2932 29.8542V27.4467C16.2932 26.1169 15.2655 25.0388 14 25.0388V22.6298C15.2675 22.6298 16.2932 21.5513 16.2932 20.2219Z" fill="#E5AE43"/>
-<path d="M30.0506 20.2233V17.8154C30.0506 16.4856 29.0228 15.4075 27.7563 15.4075V13C30.2889 13 32.3438 15.1551 32.3438 17.8154L32.3448 20.2233C32.3448 21.5532 33.3716 22.6309 34.6381 22.6309V25.0388L34.637 25.0399C33.3705 25.0399 32.3438 26.1179 32.3438 27.4478V29.8531C32.3438 32.5113 30.2913 34.67 27.7577 34.67V32.261C29.0242 32.261 30.0506 31.1829 30.0506 29.8531V27.4478C30.0506 26.0006 30.6711 24.7177 31.6328 23.835C30.6711 22.9519 30.0506 21.6698 30.0506 20.2233Z" fill="#E5AE43"/>
+<svg width="48" height="48" viewBox="0 0 48 48" fill="none" xmlns="http://www.w3.org/2000/svg">
+<g clip-path="url(#clip0_116_4)">
+<path d="M44 0H4C1.79086 0 0 1.79086 0 4V44C0 46.2091 1.79086 48 4 48H44C46.2091 48 48 46.2091 48 44V4C48 1.79086 46.2091 0 44 0Z" fill="#E5AE43" fill-opacity="0.08"/>
+<path d="M20 30L14 24L20 18L21.425 19.425L16.825 24.025L21.4 28.6L20 30ZM28 30L26.575 28.575L31.175 23.975L26.6 19.4L28 18L34 24L28 30Z" fill="#E5AE43"/>
+</g>
+<defs>
+<clipPath id="clip0_116_4">
+<rect width="48" height="48" fill="white"/>
+</clipPath>
+</defs>
 </svg>
diff --git a/packages/ui/core/src/assets/img/custom/piece/code_mention.png b/packages/ui/core/src/assets/img/custom/piece/code_mention.png
index 1d8877455b..9a93c1a125 100644
Binary files a/packages/ui/core/src/assets/img/custom/piece/code_mention.png and b/packages/ui/core/src/assets/img/custom/piece/code_mention.png differ