diff --git a/src/Microsoft.SqlTools.ServiceLayer/QueryExecution/QueryExecutionService.cs b/src/Microsoft.SqlTools.ServiceLayer/QueryExecution/QueryExecutionService.cs
index d8ff0811..e9f0ba1a 100644
--- a/src/Microsoft.SqlTools.ServiceLayer/QueryExecution/QueryExecutionService.cs
+++ b/src/Microsoft.SqlTools.ServiceLayer/QueryExecution/QueryExecutionService.cs
@@ -99,7 +99,7 @@ namespace Microsoft.SqlTools.ServiceLayer.QueryExecution
         {
             // Attempt to load the query
             Query query;
-            if (!ActiveQueries.TryGetValue(subsetParams.OwnerId, out query))
+            if (!ActiveQueries.TryGetValue(subsetParams.OwnerUri, out query))
             {
                 var errorResult = new QueryExecuteSubsetResult
                 {