1 files changed, 8 insertions, 3 deletions
diff --git a/src/shared/battle/Battle/Struct/Omnimods.elm b/src/shared/battle/Battle/Struct/Omnimods.elm
index 2b3c011..e9c3089 100644
--- a/src/shared/battle/Battle/Struct/Omnimods.elm
+++ b/src/shared/battle/Battle/Struct/Omnimods.elm
@@ -195,9 +195,14 @@ scale multiplier omnimods =
          (Dict.map (scale_dict_value multiplier) omnimods.defense)
    }
 
-get_attribute_mod : String -> Type -> Int
-get_attribute_mod att_name omnimods =
-   case (Dict.get att_name omnimods.attributes) of
+get_attribute_mod : Battle.Struct.Attributes.Category -> Type -> Int
+get_attribute_mod att omnimods =
+   case
+      (Dict.get
+         (Battle.Struct.Attributes.encode_category att)
+         omnimods.attributes
+      )
+   of
       (Just e) -> e
       Nothing -> 0