diff --git a/test/Microsoft.SqlTools.ServiceLayer.Test/QueryExecution/CancelTests.cs b/test/Microsoft.SqlTools.ServiceLayer.Test/QueryExecution/CancelTests.cs
index 458e0f00..3899507c 100644
--- a/test/Microsoft.SqlTools.ServiceLayer.Test/QueryExecution/CancelTests.cs
+++ b/test/Microsoft.SqlTools.ServiceLayer.Test/QueryExecution/CancelTests.cs
@@ -18,7 +18,7 @@ namespace Microsoft.SqlTools.ServiceLayer.Test.QueryExecution
 {
     public class CancelTests
     {
-        //[Fact]
+        [Fact]
         public async void CancelInProgressQueryTest()
         {
             // Set up file for returning the query
@@ -51,8 +51,8 @@ namespace Microsoft.SqlTools.ServiceLayer.Test.QueryExecution
             VerifyQueryCancelCallCount(cancelRequest, Times.Once(), Times.Never());
             Assert.Null(result.Messages);
             
-            // ... The query should have been disposed as well
-            Assert.Empty(queryService.ActiveQueries);
+            // ... The query should not have been disposed
+            Assert.Equal(1, queryService.ActiveQueries.Count);
         }
 
         [Fact]