diff --git a/Gameboard.ShogiUI.UnitTests/PathFinding/PathFinder2DShould.cs b/Gameboard.ShogiUI.UnitTests/PathFinding/PathFinder2DShould.cs
index 089d142..983ec6f 100644
--- a/Gameboard.ShogiUI.UnitTests/PathFinding/PathFinder2DShould.cs
+++ b/Gameboard.ShogiUI.UnitTests/PathFinding/PathFinder2DShould.cs
@@ -1,5 +1,4 @@
 using FluentAssertions;
-using Gameboard.ShogiUI.BoardState.Pieces;
 using Microsoft.VisualStudio.TestTools.UnitTesting;
 using PathFinding;
 using System.Numerics;
@@ -12,22 +11,21 @@ namespace Gameboard.ShogiUI.UnitTests.PathFinding
     [TestMethod]
     public void Maths()
     {
-
-      var result = PathFinder2D<Piece>.IsPathable(
+      var result = PathFinder2D<IPlanarElement>.IsPathable(
         new Vector2(2, 2),
         new Vector2(7, 7),
         new Vector2(1, 1)
         );
       result.Should().BeTrue();
 
-      result = PathFinder2D<Piece>.IsPathable(
+      result = PathFinder2D<IPlanarElement>.IsPathable(
         new Vector2(2, 2),
         new Vector2(7, 7),
         new Vector2(0, 0)
         );
       result.Should().BeFalse();
 
-      result = PathFinder2D<Piece>.IsPathable(
+      result = PathFinder2D<IPlanarElement>.IsPathable(
         new Vector2(2, 2),
         new Vector2(7, 7),
         new Vector2(-1, 1)