diff --git a/headless-services/commons/language-server-test-harness/src/main/java/org/springframework/ide/vscode/languageserver/testharness/CodeAction.java b/headless-services/commons/language-server-test-harness/src/main/java/org/springframework/ide/vscode/languageserver/testharness/CodeAction.java
index f060ad9de..6d6edef3b 100644
--- a/headless-services/commons/language-server-test-harness/src/main/java/org/springframework/ide/vscode/languageserver/testharness/CodeAction.java
+++ b/headless-services/commons/language-server-test-harness/src/main/java/org/springframework/ide/vscode/languageserver/testharness/CodeAction.java
@@ -53,5 +53,9 @@ public class CodeAction {
 			}
 		}
 	}
+	
+	public Command getCommand() {
+		return ca.isLeft() ? ca.getLeft() : ca.getRight().getCommand();
+	}
 
 }
diff --git a/headless-services/spring-boot-language-server/src/test/java/org/springframework/ide/vscode/boot/test/ApplicationYamlEditorTest.java b/headless-services/spring-boot-language-server/src/test/java/org/springframework/ide/vscode/boot/test/ApplicationYamlEditorTest.java
index bf4a03c28..321fe0dbb 100644
--- a/headless-services/spring-boot-language-server/src/test/java/org/springframework/ide/vscode/boot/test/ApplicationYamlEditorTest.java
+++ b/headless-services/spring-boot-language-server/src/test/java/org/springframework/ide/vscode/boot/test/ApplicationYamlEditorTest.java
@@ -24,6 +24,7 @@ import java.util.List;
 import java.util.Optional;
 
 import org.apache.commons.io.FileUtils;
+import org.eclipse.lsp4j.Command;
 import org.eclipse.lsp4j.Diagnostic;
 import org.eclipse.lsp4j.DiagnosticSeverity;
 import org.junit.After;
@@ -3286,9 +3287,10 @@ public class ApplicationYamlEditorTest extends AbstractPropsEditorTest {
 	}
 	
 	private String getMissingPropertyName(CodeAction fix) {
-		String cmd = fix.command.getCommand();
+		Command command = fix.getCommand();
+		String cmd = command.getCommand();
 		assertEquals("sts.vscode-spring-boot.codeAction", cmd);
-		List<Object> args = fix.command.getArguments();
+		List<Object> args = command.getArguments();
 		assertEquals("MISSING_PROPERTY_APP", args.get(0));
 		MissingPropertyData data = (MissingPropertyData) args.get(1);
 		return data.getProperty();