Fix problem with the UI in workflow_viewer