diff --git a/nifi/nifi-nar-bundles/nifi-framework-bundle/nifi-framework/nifi-web/nifi-web-ui/src/main/webapp/js/nf/canvas/nf-canvas-toolbox.js b/nifi/nifi-nar-bundles/nifi-framework-bundle/nifi-framework/nifi-web/nifi-web-ui/src/main/webapp/js/nf/canvas/nf-canvas-toolbox.js
index 8d005e2f0a..325564177f 100644
--- a/nifi/nifi-nar-bundles/nifi-framework-bundle/nifi-framework/nifi-web/nifi-web-ui/src/main/webapp/js/nf/canvas/nf-canvas-toolbox.js
+++ b/nifi/nifi-nar-bundles/nifi-framework-bundle/nifi-framework/nifi-web/nifi-web-ui/src/main/webapp/js/nf/canvas/nf-canvas-toolbox.js
@@ -882,7 +882,9 @@ nf.CanvasToolbox = (function () {
                 nf.Client.setRevision(response.revision);
 
                 // add the label to the graph
-                nf.Label.add(response.label, true);
+                nf.Graph.add({
+                    'labels': [response.label]
+                }, true);
 
                 // update the birdseye
                 nf.Birdseye.refresh();