2 files changed, 11 insertions, 2 deletions

diff --git a/src/bin/client/games/manager.rs b/src/bin/client/games/manager.rs
index 6959214..0b5b4cb 100644
--- a/src/bin/client/games/manager.rs
+++ b/src/bin/client/games/manager.rs
@@ -55,7 +55,7 @@ impl GameManager {
     ) -> Result<(), GameError> {
         let game_type = self
             .pending_game_type
-            .take()
+            .as_ref()
             .ok_or(GameError::InvalidAction)?;
 
         let state = self
@@ -63,7 +63,7 @@ impl GameManager {
             .as_mut()
             .ok_or(GameError::InvalidAction)?;
 
-        if state.phase != GamePhase::WaitingToStart {
+        if state.phase != GamePhase::WaitingToStart && state.phase != GamePhase::Finished {
             return Err(GameError::InvalidAction);
         }
 
@@ -103,6 +103,13 @@ impl GameManager {
             .unwrap_or(false)
     }
 
+    pub fn reset_phase_to_waiting(&mut self) {
+        if let Some(ref mut session) = self.current_session {
+            session.phase = GamePhase::WaitingToStart;
+            session.data.clear();
+        }
+    }
+
     pub fn end_game(&mut self) {
         self.current_session = None;
         self.pending_game_type = None;
diff --git a/src/bin/client/games/traits.rs b/src/bin/client/games/traits.rs
index 19f2638..aed0c24 100644
--- a/src/bin/client/games/traits.rs
+++ b/src/bin/client/games/traits.rs
@@ -9,6 +9,8 @@ pub enum GameError {
 #[derive(Debug, Clone, PartialEq)]
 pub enum GamePhase {
     WaitingToStart,
+    Playing,
+    Finished,
 }
 
 #[derive(Debug, Clone)]