diff --git a/src/AI/Affinities.py b/src/AI/Affinities.py
index b35d505..01ec59e 100644
--- a/src/AI/Affinities.py
+++ b/src/AI/Affinities.py
@@ -24,5 +24,5 @@ def pickWeightedAffinity(affinities: Affinities):
     population = [Classifiers.FOOD, Classifiers.WATER, Classifiers.REST]
     weights = affinities.getWeigths()
 
-    return random.choices(population, weights)
+    return random.choices(population, weights)[0]