1

Assets/Artists/Scripts/Entities/Melee/EntityMelee.cs

@@ -9,7 +9,7 @@ using UnityEngine;
 namespace BITKit.Entities
 {
     [CustomType(typeof(IMeleeCombat))]
-    public class EntityMelee : EntityComponent,IMeleeCombat
+    public class EntityMelee : EntityBehavior,IMeleeCombat
     {
         [SerializeField] private UnityAnimator unityAnimator;
         
@@ -21,8 +21,8 @@ namespace BITKit.Entities
         [Inject(true)] private IEntityOverride entityOverride;
         public override void OnStart()
         {
-            entity.AddListener<int>("Melee", Melee);
-            entity.AddListener<string>(AIAction);
+            UnityEntity.AddListener<int>("Melee", Melee);
+            UnityEntity.AddListener<string>(AIAction);
             
             unityAnimator[0].onStateEnter += OnStateEnter;
         }