diff --git a/static/js/entityDialog.js b/static/js/entityDialog.js
index 66081bcc..fe0b8487 100644
--- a/static/js/entityDialog.js
+++ b/static/js/entityDialog.js
@@ -17,7 +17,9 @@ pandora.ui.entityDialog = function() {
                 })
                 .bindEvent({
                     click: function() {
+                        var part = ui.part.entities;
                         that.close();
+                        ui.set({'part.entities': part});
                         pandora.$ui.entitiesDialog = pandora.ui.entitiesDialog().open();
                     }
                 }),