diff --git a/src/test/java/dotsandboxes/DotsAndBoxesGridTest.java b/src/test/java/dotsandboxes/DotsAndBoxesGridTest.java
index 75fa3c2acceea84772f7e83cc8b063b6ccbaa27f..ad3402038d79b6ed01636426ef52ab2f72bf87be 100644
--- a/src/test/java/dotsandboxes/DotsAndBoxesGridTest.java
+++ b/src/test/java/dotsandboxes/DotsAndBoxesGridTest.java
@@ -32,7 +32,7 @@ public class DotsAndBoxesGridTest {
     public void testBoxCompletionDetection() {
         DotsAndBoxesGrid grid = new DotsAndBoxesGrid(4, 3, 2);
         
-        / Draw only three sides of a box (not completing it)
+        // Draw only three sides of a box (not completing it)
         grid.drawHorizontal(0, 0, 1); // Top
         grid.drawVertical(0, 0, 1);   // Left
         grid.drawHorizontal(0, 1, 1); // Bottom
@@ -41,26 +41,23 @@ public class DotsAndBoxesGridTest {
         assertTrue(grid.boxComplete(0, 0), "The box should not be completed.");
     }
 
-    @Test
-    public void testBoxComplete() {
-        // Draw lines around a box and check if it's completed
-        game.drawHorizontal(0, 0, 1); // Top horizontal line
-        game.drawVertical(0, 0, 1); // Left vertical line
-        game.drawHorizontal(0, 1, 1); // Bottom horizontal line
-        game.drawVertical(1, 0, 1); // Right vertical line
 
-        assertTrue("Box should be completed", game.boxComplete(0, 0));
-    }
 
-    @Test(expected = IllegalStateException.class)
+    @Test
     public void testDrawHorizontalLineAlreadyDrawn() {
-        game.drawHorizontal(0, 0, 1);
-        game.drawHorizontal(0, 0, 1); // Drawing the same line should throw an exception
+        DotsAndBoxesGrid grid = new DotsAndBoxesGrid(4,4,2);
+        grid.drawHorizontal(0, 0, 1);
+        assertThrows(IllegalStateException.class, () -> {
+            grid.drawHorizontal(0, 0, 1); // Drawing the same line should throw an exception
+        });
     }
 
-    @Test(expected = IllegalStateException.class)
+    @Test
     public void testDrawVerticalLineAlreadyDrawn() {
-        game.drawVertical(0, 0, 1);
-        game.drawVertical(0, 0, 1); // Drawing the same line should throw an exception
+        DotsAndBoxesGrid grid = new DotsAndBoxesGrid(4,4,2);
+        grid.drawVertical(0, 0, 1);
+        assertThrows(IllegalStateException.class, () -> {
+            grid.drawVertical(0, 0, 1); // Drawing the same line should throw an exception
+        });    
     }
 }