X-Git-Url: https://git.arvados.org/arvados-workbench2.git/blobdiff_plain/f17af4bfbdd3b0054d3494cdc6c5f75c2e872d9f..d87ff106a9f51defe5aafab2ad23dc1dee267b29:/src/models/workflow.ts

diff --git a/src/models/workflow.ts b/src/models/workflow.ts
index ad84bd9e..6d21dbc7 100644
--- a/src/models/workflow.ts
+++ b/src/models/workflow.ts
@@ -139,7 +139,7 @@ export const parseWorkflowDefinition = (workflow: WorkflowResource): WorkflowRes
 
 export const getWorkflow = (workflowDefinition: WorkflowResourceDefinition) => {
     if (!workflowDefinition.$graph) { return undefined; }
-    const mainWorkflow = workflowDefinition.$graph.find(item => item.class === 'Workflow' && item.id === '#main');
+    const mainWorkflow = workflowDefinition.$graph.find(item => item.id === '#main');
     return mainWorkflow
         ? mainWorkflow
         : undefined;
@@ -153,7 +153,7 @@ export const getWorkflowInputs = (workflowDefinition: WorkflowResourceDefinition
 };
 
 export const getInputLabel = (input: CommandInputParameter) => {
-    return `${input.label || input.id}`;
+    return `${input.label || input.id.split('/').pop()}`;
 };
 
 export const isRequiredInput = ({ type }: CommandInputParameter) => {