diff --git a/tests/test_Cell.py b/tests/test_Cell.py
index 9f37042..7b0561a 100644
--- a/tests/test_Cell.py
+++ b/tests/test_Cell.py
@@ -1,4 +1,3 @@
-import pytest
 from amaz_lib.Cell import Cell
 
 
diff --git a/tests/test_MazeSolver.py b/tests/test_MazeSolver.py
index b8f2f4c..4ec0061 100644
--- a/tests/test_MazeSolver.py
+++ b/tests/test_MazeSolver.py
@@ -1,6 +1,6 @@
 from amaz_lib.Cell import Cell
 import numpy as np
-from amaz_lib import AStar, Maze, MazeSolver
+from amaz_lib import AStar, Maze
 
 
 def test_solver() -> None: