diff --git a/src/server/decisions.rs b/src/server/decisions.rs
index 83e5f89..23d48b3 100644
--- a/src/server/decisions.rs
+++ b/src/server/decisions.rs
@@ -220,26 +220,12 @@ pub async fn create_decision(
                                 score,
                                 confidence,
                                 probabilities: {
-                                    let reverse_legend = HashMap::<_, _, RandomState>::from_iter(
-                                        legend.into_iter().map(|(index, value)| {
-                                            (
-                                                value,
-                                                index
-                                                    .parse::<i32>()
-                                                    .expect("this should be an integer, I believe"),
-                                            )
-                                        }),
-                                    );
-
                                     probabilities
                                         .into_iter()
                                         .map(|(choice_id, probability)| {
                                             openai::decisions::ScoreProbability {
-                                                value: reverse_legend
-                                                    .get(&choice_id)
-                                                    .expect("it should exist, I think")
-                                                    .clone(),
-                                                label: choice_id,
+                                                value: choice_id.parse().expect("it should be a number, I think"),
+                                                label: legend.get(&choice_id).expect("it should exist, I think").clone(),
                                                 probability,
                                             }
                                         })