diff --git a/server/gx-workflow-ls-format2/tests/unit/schema.test.ts b/server/gx-workflow-ls-format2/tests/unit/schema.test.ts
index 3834f83..6dcea71 100644
--- a/server/gx-workflow-ls-format2/tests/unit/schema.test.ts
+++ b/server/gx-workflow-ls-format2/tests/unit/schema.test.ts
@@ -22,13 +22,13 @@ describe("Gxformat2 Schema Handling", () => {
         it.each([
           [["class"], "class"],
           [["inputs"], "inputs"],
-          [["inputs", 0], "InputParameter"],
+          [["inputs", 0], "WorkflowInputParameter"],
           [["inputs", 0, "doc"], "doc"],
-          [["inputs", "input1"], "InputParameter"],
+          [["inputs", "input1"], "WorkflowInputParameter"],
           [["inputs", "input1", "label"], "label"],
           [["outputs"], "outputs"],
-          [["outputs", 1], "OutputParameter"],
-          [["outputs", "output1"], "OutputParameter"],
+          [["outputs", 1], "WorkflowOutputParameter"],
+          [["outputs", "output1"], "WorkflowOutputParameter"],
           [["steps"], "steps"],
           [["steps", 5], "WorkflowStep"],
           [["steps", "random", "tool_id"], "tool_id"],