Made it able to update the sprite on screen.

MrBigsock/Assets/Code/UI/AbilityElement.cs

@@ -29,8 +29,10 @@ namespace BigSock.UI
 
 
     public AbilityElement WithAbility(AbilityEntity ability) {
+      _sprite ??= transform.Find("Sprite").GetComponent<UnityEngine.UI.Image>();
+
       if (_sprite != null)
-        _sprite.overrideSprite = ability.Ability.Icon ?? _sprite.overrideSprite;
+          _sprite.overrideSprite = ability.Ability.Icon ?? _sprite.overrideSprite;
       return this;
     }
 
@@ -39,7 +41,7 @@ namespace BigSock.UI
     void Start() {
       chargeSlider = transform.Find("ChargeSlider").GetComponent<Slider>();
       cooldownSlider = transform.Find("ReloadSlider").GetComponent<Slider>();
-      _sprite = transform.Find("Sprite").GetComponent<UnityEngine.UI.Image>();
+      _sprite ??= transform.Find("Sprite").GetComponent<UnityEngine.UI.Image>();
     }
 
     // Update is called once per frame