fix: fix abilities selecting the right cooldown

1 files changed, 5 insertions, 5 deletions

diff --git a/src/shared/champion.rs b/src/shared/champion.rs
index 8de12a5..28d1ff2 100644
--- a/src/shared/champion.rs
+++ b/src/shared/champion.rs
@@ -40,14 +40,14 @@ impl Stats {
 }
 
 impl Champion {
-    pub fn to_ability(self, attack_key: AttackKey) -> Ability {
+    pub fn to_ability(self, ability_slot: AbilitySlot) -> Ability {
         match self {
-            Champion::Meele => match attack_key {
-                AttackKey::Q => Ability::Directional(DirectionalAbility::Spear),
+            Champion::Meele => match ability_slot {
+                AbilitySlot::Q => Ability::Directional(DirectionalAbility::Spear),
                 _ => Ability::Targeted(TargetedAbility::MeeleAttack),
             },
-            Champion::Ranged => match attack_key {
-                AttackKey::Q => Ability::Directional(DirectionalAbility::Spear),
+            Champion::Ranged => match ability_slot {
+                AbilitySlot::Q => Ability::Directional(DirectionalAbility::Spear),
                 _ => Ability::Targeted(TargetedAbility::RangedAttack),
             },
         }