Fix AbilityScoreCollection: should not contain vectors

1 files changed, 4 insertions, 7 deletions

diff --git a/src/rules/ability_scores.rs b/src/rules/ability_scores.rs
index 205c1a3..2b9acf0 100644
--- a/src/rules/ability_scores.rs
+++ b/src/rules/ability_scores.rs
@@ -26,7 +26,7 @@ impl AbilityScore {
 
 #[derive(Debug)]
 pub struct AbilityScoreCollection {
-    scores: HashMap<AbilityScore, Vec<u32>>,
+    scores: HashMap<AbilityScore, u32>,
 }
 
 impl AbilityScoreCollection {
@@ -36,14 +36,11 @@ impl AbilityScoreCollection {
         }
     }
 
-    pub fn add_score(&mut self, ability_score: AbilityScore, bonus: u32) {
-        self.scores
-            .entry(ability_score)
-            .or_insert_with(Vec::new)
-            .push(bonus);
+    pub fn add_score(&mut self, ability_score: AbilityScore, value: u32) {
+        self.scores.insert(ability_score, value);
     }
 
-    pub fn get_score(&self, ability_score: AbilityScore) -> Option<&Vec<u32>> {
+    pub fn get_score(&self, ability_score: AbilityScore) -> Option<&u32> {
         self.scores.get(&ability_score)
     }
 }