Only match the good moves

2 files changed, 2 insertions, 2 deletions

diff --git a/src/lib.rs b/src/lib.rs
index 519b804..c36a817 100644
--- a/src/lib.rs
+++ b/src/lib.rs
@@ -128,7 +128,7 @@ fn shortest_path_first_command(initial_state: &GameState) -> Option<Command> {
         .zip(shortest_path_states.0.iter().skip(1))
         .map(|(state, next)| {
             let player_move = state
-                .valid_moves(0)
+                .good_moves(0)
                 .into_iter()
                 .filter(|player_move| *player_move != Command::UseOil)
                 .find(|player_move| {
diff --git a/src/state.rs b/src/state.rs
index be42228..acbce80 100644
--- a/src/state.rs
+++ b/src/state.rs
@@ -231,7 +231,7 @@ impl GameState {
 
     pub fn good_moves(&self, player_index: usize) -> Vec<Command> {
         let player = &self.players[player_index];
-        let mut result = Vec::with_capacity(7);
+        let mut result = Vec::with_capacity(5);
         result.push(Command::Accelerate);
         if player.position.y > MIN_Y {
             result.push(Command::TurnLeft);