From bf71b32c82b29d8f8398ea029fcd4f6b6b073566 Mon Sep 17 00:00:00 2001
From: Anuken <arnukren@gmail.com>
Date: Thu, 10 Feb 2022 16:02:13 -0500
Subject: [PATCH] Fixed lacking AI (?)

---
 core/src/mindustry/type/unit/ErekirUnitType.java | 3 ++-
 1 file changed, 2 insertions(+), 1 deletion(-)

diff --git a/core/src/mindustry/type/unit/ErekirUnitType.java b/core/src/mindustry/type/unit/ErekirUnitType.java
index 099ab679b8..c29d121523 100644
--- a/core/src/mindustry/type/unit/ErekirUnitType.java
+++ b/core/src/mindustry/type/unit/ErekirUnitType.java
@@ -1,5 +1,6 @@
 package mindustry.type.unit;
 
+import mindustry.*;
 import mindustry.ai.types.*;
 import mindustry.graphics.*;
 import mindustry.type.*;
@@ -14,6 +15,6 @@ public class ErekirUnitType extends UnitType{
         outlineColor = Pal.darkOutline;
         envDisabled = Env.space;
         coreUnitDock = true;
-        unitBasedDefaultController = u -> !playerControllable || u.team.isAI() ? defaultController.get() : new CommandAI();
+        unitBasedDefaultController = u -> !playerControllable || u.team.isAI() || (Vars.state.rules.attackMode && u.team == Vars.state.rules.waveTeam) ? defaultController.get() : new CommandAI();
     }
 }