diff --git a/Directory.Packages.props b/Directory.Packages.props
index 77687d45f7..2548b88d72 100644
--- a/Directory.Packages.props
+++ b/Directory.Packages.props
@@ -21,11 +21,10 @@
-
+
-
-
+
diff --git a/test/AssemblyInfo.cs b/test/AssemblyInfo.cs
index c7fc3b1f90..4647e4471c 100644
--- a/test/AssemblyInfo.cs
+++ b/test/AssemblyInfo.cs
@@ -1 +1,4 @@
-[assembly: CollectionBehavior(DisableTestParallelization = true)]
+using ExRam.Gremlinq.Tests.Infrastructure;
+
+[assembly: CollectionBehavior(DisableTestParallelization = true)]
+[assembly: TestFramework(typeof(GremlinqTestFramework))]
diff --git a/test/Core.AspNet.Tests/ExRam.Gremlinq.Core.AspNet.Tests.csproj b/test/Core.AspNet.Tests/ExRam.Gremlinq.Core.AspNet.Tests.csproj
index 707a54a833..f9170707ab 100644
--- a/test/Core.AspNet.Tests/ExRam.Gremlinq.Core.AspNet.Tests.csproj
+++ b/test/Core.AspNet.Tests/ExRam.Gremlinq.Core.AspNet.Tests.csproj
@@ -11,6 +11,7 @@
+
diff --git a/test/Core.Tests/Assumptions.cs b/test/Core.Tests/Assumptions.cs
index b736e493a5..37be526de8 100644
--- a/test/Core.Tests/Assumptions.cs
+++ b/test/Core.Tests/Assumptions.cs
@@ -11,7 +11,7 @@ public async Task SemaphoreSlim_WaitAsync_fails_upon_cancellation()
var cts = new CancellationTokenSource();
var waitTask = semaphore.WaitAsync(cts.Token);
- await Task.Delay(100);
+ await Task.Delay(100, TestContext.Current.CancellationToken);
waitTask.IsCompleted
.Should()
diff --git a/test/Core.Tests/GremlinQueryExecutorTest.cs b/test/Core.Tests/GremlinQueryExecutorTest.cs
index 1298089333..9c71294748 100644
--- a/test/Core.Tests/GremlinQueryExecutorTest.cs
+++ b/test/Core.Tests/GremlinQueryExecutorTest.cs
@@ -24,7 +24,7 @@ public void Invalid()
GremlinQueryExecutor.Invalid
.Execute