diff --git a/src/main/java/com/alttd/ctf/commands/subcommands/SelectClass.java b/src/main/java/com/alttd/ctf/commands/subcommands/SelectClass.java index 7e68ae9..49ee4e6 100644 --- a/src/main/java/com/alttd/ctf/commands/subcommands/SelectClass.java +++ b/src/main/java/com/alttd/ctf/commands/subcommands/SelectClass.java @@ -5,6 +5,7 @@ import com.alttd.ctf.config.Messages; import com.alttd.ctf.game.GameManager; import com.alttd.ctf.game.GamePhase; import com.alttd.ctf.game_class.GameClass; +import com.alttd.ctf.game_class.creation.EngineerCreator; import com.alttd.ctf.game_class.creation.FighterCreator; import com.alttd.ctf.game_class.creation.TankCreator; import com.alttd.ctf.gui.ClassSelectionGUI; @@ -28,6 +29,7 @@ public class SelectClass extends SubCommand { gameManager.getTeams().forEach(team -> { gameClasses.computeIfAbsent(team.getId(), teamId -> new ArrayList<>()).add(FighterCreator.createFighter(team.getColor())); gameClasses.computeIfAbsent(team.getId(), teamId -> new ArrayList<>()).add(TankCreator.createTank(team.getColor())); + gameClasses.computeIfAbsent(team.getId(), teamId -> new ArrayList<>()).add(EngineerCreator.createEngineer(team.getColor())); }); }