From 34c5e448ce55d2d5287ea07e28f16076a678a55a Mon Sep 17 00:00:00 2001
From: GarandPLG <garandplg@garandplg.com>
Date: Mon, 18 May 2026 17:37:31 +0200
Subject: [PATCH] .

---
 src/app/states/skirmish_states/board.rs | 22 +++++++++++-----------
 1 file changed, 11 insertions(+), 11 deletions(-)

diff --git a/src/app/states/skirmish_states/board.rs b/src/app/states/skirmish_states/board.rs
index 8eddd5f..754e510 100644
--- a/src/app/states/skirmish_states/board.rs
+++ b/src/app/states/skirmish_states/board.rs
@@ -11,7 +11,6 @@ use crate::app::{
     },
     widgets::CellWidget,
 };
-use log::info;
 use ratatui::layout::Rect;
 use std::collections::VecDeque;
 
@@ -141,12 +140,12 @@ impl BoardState {
         self.marked_cells
             .marked_cells
             .iter()
-            .map(move |&(row, col)| &self.cells[row][col])
+            .map(|&(row, col)| &self.cells[row][col])
             .collect()
     }
 
-    pub fn get_marked_cells(&self) -> VecDeque<(usize, usize)> {
-        self.marked_cells.marked_cells.clone()
+    pub fn get_marked_cells(&self) -> &VecDeque<(usize, usize)> {
+        &self.marked_cells.marked_cells
     }
 
     pub fn toggle_marking(&mut self) {
@@ -170,7 +169,7 @@ impl BoardState {
             .marked_cells
             .push_back((new_cell.0, new_cell.1));
 
-        info!("{:?}", self.marked_cells);
+        // info!("{:?}", self.marked_cells.marked_cells);
     }
 
     pub fn undo_marked_cell(&mut self) {
@@ -220,13 +219,14 @@ impl BoardState {
 
         let (row, col) = self.marked_cells.selected_unit.get_coords();
 
-        // if !self.marked_cells.marked_cells.len() == 1 {
-        let task: Tasks = Tasks::Digging(DiggingTask::new(self.marked_cells.marked_cells.clone()));
+        if self.marked_cells.marked_cells.len() > 1 {
+            let task: Tasks =
+                Tasks::Digging(DiggingTask::new(self.marked_cells.marked_cells.clone()));
 
-        self.get_mut_cell(row, col)
-            .get_mut_option_unit()
-            .set_task(task);
-        // }
+            self.get_mut_cell(row, col)
+                .get_mut_option_unit()
+                .set_task(task);
+        }
 
         self.marked_cells.selected_unit = None;
         self.marked_cells.marked_cells.clear();