diff --git a/tpot2/individual_representations/graph_pipeline_individual/individual.py b/tpot2/individual_representations/graph_pipeline_individual/individual.py
index bc2a883a..f890e80f 100644
--- a/tpot2/individual_representations/graph_pipeline_individual/individual.py
+++ b/tpot2/individual_representations/graph_pipeline_individual/individual.py
@@ -1193,7 +1193,10 @@ def random_weighted_sort(l,weights, rng_=None):
     sorted_l = []
     indeces = {i: weights[i] for i in range(len(l))}
     while len(indeces) > 0:
-        next_item = rng.choice(list(indeces.keys()), p=list(indeces.values()))
+        keys = list(indeces.keys())
+        p = np.array([indeces[k] for k in keys])
+        p = p / p.sum()
+        next_item = rng.choice(list(indeces.keys()), p=p)
         indeces.pop(next_item)
         sorted_l.append(l[next_item])