From 82494091cc807b91a86979389d9b3b2110c67185 Mon Sep 17 00:00:00 2001
From: Darguima <dsgdevbraga@gmail.com>
Date: Sat, 20 Feb 2021 12:12:48 +0000
Subject: [PATCH] Corrected the function - sortAShufleFromCasesArray - for when
 only have 0/1 cases selected

---
 src/TrainPage/index.tsx | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/src/TrainPage/index.tsx b/src/TrainPage/index.tsx
index 1ce3b28..e095175 100644
--- a/src/TrainPage/index.tsx
+++ b/src/TrainPage/index.tsx
@@ -64,7 +64,7 @@ const TrainPage:React.FC<TrainPageProps> = ({ route: { params: { methodPhase } }
   }, [])
 
   const sortAShufleFromCasesArray = async () => {
-    if (casesArray.unsolved.length === 1) {
+    if (casesArray.unsolved.length === 1 && casesArray.solved.length !== 0) {
       const newUnsolvedCasesArray: Array<fridrichCaseSchema> = casesArray.unsolved
       const newSolvedCasesArray: Array<fridrichCaseSchema> = casesArray.solved
 
@@ -85,7 +85,7 @@ const TrainPage:React.FC<TrainPageProps> = ({ route: { params: { methodPhase } }
         setCasesArray({ unsolved: newUnsolvedCasesArray, solved: newSolvedCasesArray })
       }
 
-      setCaseOnScreen(casesArray.unsolved[0])
+      setCaseOnScreen(casesArray.unsolved[0] || { name: 'Final', shuffle: 'Final', solve: 'Final', solved: true })
     }
   }