From a30d4a4514d73ddcc9f2e948a7820ad5a1e16eb8 Mon Sep 17 00:00:00 2001
From: Khopa <clemguip@gmail.com>
Date: Wed, 3 Jul 2019 21:06:03 +0200
Subject: [PATCH] Fix in 'unit_task' method

---
 game/db.py | 6 ++++--
 1 file changed, 4 insertions(+), 2 deletions(-)

diff --git a/game/db.py b/game/db.py
index d05f3f09..9ce1b577 100644
--- a/game/db.py
+++ b/game/db.py
@@ -940,8 +940,10 @@ def unit_task(unit: UnitType) -> Task:
     for task, units in UNIT_BY_TASK.items():
         if unit in units:
             return task
-        elif unit in SAM_CONVERT:
-            return SAM_CONVERT[unit]
+
+    if unit in SAM_CONVERT:
+        return unit_task(SAM_CONVERT[unit])
+
     assert False