Fixed NPE, this fix is required for workflow plotting.