diff --git a/test/MLOps.NET.Tests/MLOps.NET.Tests.csproj b/test/MLOps.NET.Tests/MLOps.NET.Tests.csproj
index 6169fa62..b88a20e6 100644
--- a/test/MLOps.NET.Tests/MLOps.NET.Tests.csproj
+++ b/test/MLOps.NET.Tests/MLOps.NET.Tests.csproj
@@ -8,7 +8,7 @@
     <PackageReference Include="FluentAssertions" Version="6.2.0" />
     <PackageReference Include="Microsoft.EntityFrameworkCore.InMemory" Version="5.0.11" />
     <PackageReference Include="Microsoft.ML" Version="1.6.0" />
-    <PackageReference Include="Microsoft.ML.FastTree" Version="1.6.0" />
+    <PackageReference Include="Microsoft.ML.FastTree" Version="2.0.1" />
     <PackageReference Include="Microsoft.NET.Test.Sdk" Version="17.0.0" />
     <PackageReference Include="MSTest.TestAdapter" Version="2.2.7" />
     <PackageReference Include="MSTest.TestFramework" Version="2.2.7" />