diff --git a/static/js/entityDialog.js b/static/js/entityDialog.js
index 6dade0e73..90123e72d 100644
--- a/static/js/entityDialog.js
+++ b/static/js/entityDialog.js
@@ -17,10 +17,16 @@ pandora.ui.entityDialog = function() {
                 })
                 .bindEvent({
                     click: function() {
-                        var part = ui.part.entities;
+                        var id = [ui.part.entities];
                         that.close();
-                        pandora.UI.set({'part.entities': part});
-                        // FIXME: Also set ui.entitiesType
+                        pandora.api.getEntity({
+                            id: id
+                        }, function(result) {
+                            var type = result.data.type;
+                            pandora.UI.set(Ox.extend({
+                                entitiesType: type
+                            }, 'entitiesSelection.' + type, [id]));
+                        });
                         pandora.$ui.entitiesDialog = pandora.ui.entitiesDialog().open();
                     }
                 }),