diff --git a/Source/Editor/CustomEditors/Editors/ActorTransformEditor.cs b/Source/Editor/CustomEditors/Editors/ActorTransformEditor.cs
index a824076f7..fb7a0b29d 100644
--- a/Source/Editor/CustomEditors/Editors/ActorTransformEditor.cs
+++ b/Source/Editor/CustomEditors/Editors/ActorTransformEditor.cs
@@ -79,7 +79,6 @@ namespace FlaxEditor.CustomEditors.Editors
         public class ScaleEditor : Float3Editor
         {
             private Button _linkButton;
-            private SpriteBrush _linkBrush;
 
             /// <inheritdoc />
             public override void Initialize(LayoutElementsContainer layout)
@@ -89,20 +88,16 @@ namespace FlaxEditor.CustomEditors.Editors
                 LinkValues = Editor.Instance.Windows.PropertiesWin.ScaleLinked;
 
                 // Add button with the link icon.
-                _linkBrush = new SpriteBrush(Editor.Instance.Icons.Link32);
-                _linkButton = new Button
+                //Editor.Instance.Icons.Link32
+                _linkButton = new IconButton(Editor.Instance.Icons.Link32)
                 {
                     Parent = LinkedLabel,
                     Width = 18,
                     Height = 18,
-                    BackgroundBrush = _linkBrush,
-                    AnchorPreset = AnchorPresets.TopLeft,
+                    AnchorPreset = AnchorPresets.TopLeft
                 };
 
                 _linkButton.Clicked += ToggleLink;
-                _linkButton.BorderColor = Color.Transparent;
-                _linkButton.BorderColorSelected = Color.Transparent;
-                _linkButton.BorderColorHighlighted = Color.Transparent;
                 SetLinkStyle();
 
                 var x = LinkedLabel.Text.Value.Length * 7 + 5;
@@ -142,9 +137,7 @@ namespace FlaxEditor.CustomEditors.Editors
             private void SetLinkStyle()
             {
                 Color backgroundColor = LinkValues ? FlaxEngine.GUI.Style.Current.BackgroundSelected : FlaxEngine.GUI.Style.Current.ForegroundDisabled;
-                _linkButton.BackgroundColor = backgroundColor;
-                _linkButton.BackgroundColorHighlighted = backgroundColor.RGBMultiplied(0.9f);
-                _linkButton.BackgroundColorSelected = backgroundColor.RGBMultiplied(0.8f);
+                _linkButton.SetColors(backgroundColor);
                 _linkButton.TooltipText = (LinkValues ? "Unlink" : "Link") + " values for uniform scaling.";
             }
         }
diff --git a/Source/Engine/UI/GUI/Common/Button.cs b/Source/Engine/UI/GUI/Common/Button.cs
index 0e0d29615..b6ee13e41 100644
--- a/Source/Engine/UI/GUI/Common/Button.cs
+++ b/Source/Engine/UI/GUI/Common/Button.cs
@@ -10,7 +10,7 @@ namespace FlaxEngine.GUI
     public class Button : ContainerControl
     {
         /// <summary>
-        /// The default height fro the buttons.
+        /// The default height for the buttons.
         /// </summary>
         public const float DefaultHeight = 24.0f;
 
@@ -195,7 +195,7 @@ namespace FlaxEngine.GUI
         /// Sets the button colors palette based on a given main color.
         /// </summary>
         /// <param name="color">The main color.</param>
-        public void SetColors(Color color)
+        public virtual void SetColors(Color color)
         {
             BackgroundColor = color;
             BorderColor = color.RGBMultiplied(0.5f);
diff --git a/Source/Engine/UI/GUI/Common/IconButton.cs b/Source/Engine/UI/GUI/Common/IconButton.cs
index 196b4b9c4..e2652cd9f 100644
--- a/Source/Engine/UI/GUI/Common/IconButton.cs
+++ b/Source/Engine/UI/GUI/Common/IconButton.cs
@@ -9,159 +9,66 @@ namespace FlaxEngine.GUI
     /// </summary>
     public class IconButton : Button
     {
-        /// <inheritdoc />
-        public override void ClearState()
+        /// <summary>
+        /// The sprite rendered on the button.
+        /// </summary>
+        public SpriteHandle ButtonSprite { get; set; }
+
+        /// <summary>
+        /// Whether or not to hide the border of the button.
+        /// </summary>
+        public bool HideBorder = true;
+
+        /// <summary>
+        /// Initializes a new instance of the <see cref="IconButton"/> class.
+        /// </summary>
+        /// <param name="buttonSprite">The sprite used by the button.</param>
+        public IconButton(SpriteHandle buttonSprite)
+        : this(0, 0, buttonSprite)
         {
-            base.ClearState();
-            
-            if (_isPressed)
-                OnPressEnd();
         }
 
-        /// <inheritdoc />
-        public override void DrawSelf()
+        /// <summary>
+        /// Initializes a new instance of the <see cref="IconButton"/> class.
+        /// </summary>
+        /// <param name="x">Position X coordinate</param>
+        /// <param name="y">Position Y coordinate</param>
+        /// <param name="buttonSprite">The sprite used by the button.</param>
+        /// <param name="width">Width</param>
+        /// <param name="height">Height</param>
+        /// <param name="hideBorder">Whether or not to hide the border.</param>
+        public IconButton(float x, float y, SpriteHandle buttonSprite, float width = 120, float height = DefaultHeight, bool hideBorder = true)
+        : base(x, y, width, height)
         {
-            // Cache data
-            Rectangle clientRect = new Rectangle(Float2.Zero, Size);
-            bool enabled = EnabledInHierarchy;
-            Color backgroundColor = BackgroundColor;
-            Color borderColor = BorderColor;
-            Color textColor = TextColor;
-            if (!enabled)
-            {
-                backgroundColor *= 0.5f;
-                borderColor *= 0.5f;
-                textColor *= 0.6f;
-            }
-            else if (_isPressed)
-            {
-                backgroundColor = BackgroundColorSelected;
-                borderColor = BorderColorSelected;
-            }
-            else if (IsMouseOver || IsNavFocused)
-            {
-                backgroundColor = BackgroundColorHighlighted;
-                borderColor = BorderColorHighlighted;
-            }
-
-            // Draw background
-            if (BackgroundBrush != null)
-                BackgroundBrush.Draw(clientRect, backgroundColor);
-            else
-                Render2D.FillRectangle(clientRect, backgroundColor);
-            Render2D.DrawRectangle(clientRect, borderColor);
-
-            // Draw text
-            Render2D.DrawText(_font?.GetFont(), TextMaterial, _text, clientRect, textColor, TextAlignment.Center, TextAlignment.Center);
+            ButtonSprite = buttonSprite;
+            BackgroundBrush = new SpriteBrush(ButtonSprite);
+            HideBorder = hideBorder;
         }
 
-        /// <inheritdoc />
-        public override void OnMouseEnter(Float2 location)
+        /// <summary>
+        /// Initializes a new instance of the <see cref="IconButton"/> class.
+        /// </summary>
+        /// <param name="location">Position</param>
+        /// <param name="size">Size</param>
+        /// <param name="buttonSprite">The sprite used by the button.</param>
+        public IconButton(Float2 location, Float2 size, SpriteHandle buttonSprite)
+        : this(location.X, location.Y, buttonSprite, size.X, size.Y)
         {
-            base.OnMouseEnter(location);
-
-            HoverBegin?.Invoke();
         }
 
-        /// <inheritdoc />
-        public override void OnMouseLeave()
+        /// <summary>
+        /// Sets the colors of the button, taking into account the <see cref="HideBorder"/> field.>
+        /// </summary>
+        /// <param name="color">The color to use.</param>
+        public override void SetColors(Color color)
         {
-            if (_isPressed)
-            {
-                OnPressEnd();
-            }
+            BackgroundColor = color;
+            BackgroundColorSelected = color.RGBMultiplied(0.8f);
+            BackgroundColorHighlighted = color.RGBMultiplied(1.2f);
 
-            HoverEnd?.Invoke();
-
-            base.OnMouseLeave();
-        }
-
-        /// <inheritdoc />
-        public override bool OnMouseDown(Float2 location, MouseButton button)
-        {
-            if (base.OnMouseDown(location, button))
-                return true;
-
-            if (button == MouseButton.Left && !_isPressed)
-            {
-                OnPressBegin();
-                return true;
-            }
-            return false;
-        }
-
-        /// <inheritdoc />
-        public override bool OnMouseUp(Float2 location, MouseButton button)
-        {
-            if (base.OnMouseUp(location, button))
-                return true;
-
-            if (button == MouseButton.Left && _isPressed)
-            {
-                OnPressEnd();
-                OnClick();
-                return true;
-            }
-            return false;
-        }
-
-        /// <inheritdoc />
-        public override bool OnTouchDown(Float2 location, int pointerId)
-        {
-            if (base.OnTouchDown(location, pointerId))
-                return true;
-
-            if (!_isPressed)
-            {
-                OnPressBegin();
-                return true;
-            }
-            return false;
-        }
-
-        /// <inheritdoc />
-        public override bool OnTouchUp(Float2 location, int pointerId)
-        {
-            if (base.OnTouchUp(location, pointerId))
-                return true;
-
-            if (_isPressed)
-            {
-                OnPressEnd();
-                OnClick();
-                return true;
-            }
-            return false;
-        }
-
-        /// <inheritdoc />
-        public override void OnTouchLeave()
-        {
-            if (_isPressed)
-            {
-                OnPressEnd();
-            }
-
-            base.OnTouchLeave();
-        }
-
-        /// <inheritdoc />
-        public override void OnLostFocus()
-        {
-            if (_isPressed)
-            {
-                OnPressEnd();
-            }
-
-            base.OnLostFocus();
-        }
-
-        /// <inheritdoc />
-        public override void OnSubmit()
-        {
-            OnClick();
-
-            base.OnSubmit();
+            BorderColor = HideBorder ? Color.Transparent : color.RGBMultiplied(0.5f);
+            BorderColorSelected = BorderColor;
+            BorderColorHighlighted = BorderColor;
         }
     }
 }