Improve class data loading

1 files changed, 2 insertions, 3 deletions

diff --git a/src/random_tables.rs b/src/random_tables.rs
index 07f0022..4411675 100644
--- a/src/random_tables.rs
+++ b/src/random_tables.rs
@@ -1,5 +1,5 @@
 use crate::dice;
-use crate::rules::classes::Classes;
+use crate::rules::classes::CLASSES;
 use include_dir::{include_dir, Dir};
 use serde::Deserialize;
 use serde_yaml;
@@ -71,9 +71,8 @@ impl RandomTables {
                             }
 
                             if let Some(class_string) = &step.class {
-                                let classes = Classes::new().unwrap();
                                 let class =
-                                    classes.class(class_string).expect("Failed to load class.");
+                                    CLASSES.get(class_string).expect("Failed to load class.");
                                 output_text.push_str(&*class.name);
                             }