From 2c2d865d7fa713c2736c74d4e946bea65f923892 Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?Micha=C5=82=20Czeka=C5=84ski?= <michal.cz2202@gmail.com>
Date: Sun, 24 May 2020 15:51:15 +0200
Subject: [PATCH] Fix wrong argument being passed to method

---
 src/AI/SurvivalDT.py | 5 +++--
 1 file changed, 3 insertions(+), 2 deletions(-)

diff --git a/src/AI/SurvivalDT.py b/src/AI/SurvivalDT.py
index 75d2b4b..e76f580 100644
--- a/src/AI/SurvivalDT.py
+++ b/src/AI/SurvivalDT.py
@@ -56,9 +56,10 @@ class SurvivalDT:
                                              dtFoods[0].distanceFromPlayer, dtWaters[0].distanceFromPlayer,
                                              dtRestPlaces[0].distanceFromPlayer)
 
-        treeDecision, choice = self.__pickEntityAfterTreeDecision__(currentSituation, self.entityPickingDecisionTree,
+        treeDecision, choice = self.__pickEntityAfterTreeDecision__(currentSituation,
                                                                     dtFoods,
-                                                                    dtRestPlaces, dtWaters)
+                                                                    dtRestPlaces,
+                                                                    dtWaters)
 
         # If the choice happens to be the same as the last one pick something else.
         if choice == map.getEntityOnCoord(player.getFacingCoord()):