diff --git a/test/CompletionExtSample/CompletionExt.cs b/test/CompletionExtSample/CompletionExt.cs
index 3d277461..9c1f52c9 100644
--- a/test/CompletionExtSample/CompletionExt.cs
+++ b/test/CompletionExtSample/CompletionExt.cs
@@ -66,7 +66,7 @@ namespace Microsoft.SqlTools.Test.CompletionExtension
             {
                 item.Command = new Command
                 {
-                    CommandStr = "vsintellicode.completionItemSelected",
+                    command = "vsintellicode.completionItemSelected",
                     Arguments = new object[] { new Dictionary<string, string> { { "IsCommit", "True" } } }
                 };
             }
diff --git a/test/Microsoft.SqlTools.ServiceLayer.IntegrationTests/LanguageServer/LanguageServiceTests.cs b/test/Microsoft.SqlTools.ServiceLayer.IntegrationTests/LanguageServer/LanguageServiceTests.cs
index 2cb2d362..1ab1ec5e 100644
--- a/test/Microsoft.SqlTools.ServiceLayer.IntegrationTests/LanguageServer/LanguageServiceTests.cs
+++ b/test/Microsoft.SqlTools.ServiceLayer.IntegrationTests/LanguageServer/LanguageServiceTests.cs
@@ -184,7 +184,7 @@ namespace Microsoft.SqlTools.ServiceLayer.IntegrationTests.LanguageServer
             //Validate the first completion item in the list is preselected
             Assert.True(completions[0].Preselect.HasValue && completions[0].Preselect.Value, "Preselect is not set properly in the first completion item by the completion extension!");
             //Validate the Command object attached to the completion item by the extension
-            Assert.True(completions[0].Command != null && completions[0].Command.CommandStr == "vsintellicode.completionItemSelected", "Command is not set properly in the first completion item by the completion extension!");
+            Assert.True(completions[0].Command != null && completions[0].Command.command == "vsintellicode.completionItemSelected", "Command is not set properly in the first completion item by the completion extension!");
 
             //clean up the temp file
             File.Delete(assemblyCopyPath);