2 files changed, 11 insertions, 5 deletions
diff --git a/server/bot/src/algos/customer.rs b/server/bot/src/algos/customer.rs
index e602addc..6a45e379 100644
--- a/server/bot/src/algos/customer.rs
+++ b/server/bot/src/algos/customer.rs
@@ -20,7 +20,9 @@ use crate::{
     BotAlgo, BotInput,
 };
 use hurrycurry_client_lib::Game;
-use hurrycurry_protocol::{glam::IVec2, DemandIndex, Message, PacketS, PlayerID, Score};
+use hurrycurry_protocol::{
+    glam::IVec2, DemandIndex, Message, PacketS, PlayerClass, PlayerID, Score,
+};
 use log::info;
 use rand::{random, seq::IndexedRandom, thread_rng};
 
@@ -124,7 +126,7 @@ impl BotAlgo for Customer {
                         .players
                         .iter()
                         .find(|(id, p)| {
-                            p.character < 0
+                            p.class == PlayerClass::Customer
                                 && **id != me
                                 && p.movement.position.distance(chair.as_vec2() + 0.5) < 1.
                         })
@@ -186,7 +188,11 @@ impl BotAlgo for Customer {
                     if !*pinned {
                         let mut pin = false;
                         game.players_spatial_index.query(pos, 3., |pid, _| {
-                            if game.players.get(&pid).map_or(false, |p| p.character >= 0) {
+                            if game
+                                .players
+                                .get(&pid)
+                                .map_or(false, |p| p.class.is_cheflike())
+                            {
                                 pin = true
                             }
                         });
diff --git a/server/bot/src/algos/frank.rs b/server/bot/src/algos/frank.rs
index 854c73bb..1e02e297 100644
--- a/server/bot/src/algos/frank.rs
+++ b/server/bot/src/algos/frank.rs
@@ -20,7 +20,7 @@ use crate::{
     BotAlgo, BotInput,
 };
 use hurrycurry_client_lib::Game;
-use hurrycurry_protocol::{glam::Vec2, Message, PacketS, PlayerID};
+use hurrycurry_protocol::{glam::Vec2, Message, PacketS, PlayerClass, PlayerID};
 use rand::{seq::IndexedRandom, thread_rng};
 
 #[derive(Default)]
@@ -116,7 +116,7 @@ fn find_chef(game: &Game, me: PlayerID) -> Option<PlayerID> {
     let chefs = game
         .players
         .iter()
-        .filter(|(i, p)| p.character >= 0 && **i != me)
+        .filter(|(i, p)| p.class == PlayerClass::Chef && **i != me)
         .map(|(i, _)| *i)
         .collect::<Vec<_>>();
     chefs.choose(&mut thread_rng()).copied()