diff --git a/src/umbraco.controls/TreePicker/BaseTreePicker.cs b/src/umbraco.controls/TreePicker/BaseTreePicker.cs
index 6659e01e93..b6c9985ed1 100644
--- a/src/umbraco.controls/TreePicker/BaseTreePicker.cs
+++ b/src/umbraco.controls/TreePicker/BaseTreePicker.cs
@@ -199,6 +199,7 @@ namespace umbraco.uicontrols.TreePicker
             ItemTitle.Style.Add(HtmlTextWriterStyle.FontWeight, "bold");
             ItemTitle.Attributes.Add("class", "treePickerTitle");      // solely for styling, e.g. with an underline or dotted border, etc.
             ButtonContainer.Controls.Add(ItemTitle);
+            ButtonContainer.Attributes.Add("class", "buttons");
             ButtonContainer.Controls.Add(new LiteralControl(" "));
             ButtonContainer.Controls.Add(new LiteralControl(" "));
 
diff --git a/src/umbraco.controls/TreePicker/BaseTreePicker.js b/src/umbraco.controls/TreePicker/BaseTreePicker.js
index 79ad9013de..646ab90c62 100644
--- a/src/umbraco.controls/TreePicker/BaseTreePicker.js
+++ b/src/umbraco.controls/TreePicker/BaseTreePicker.js
@@ -10,9 +10,12 @@
                 section: s,
                 callback: function (data) {
                     //this returns the content object picked
-                    jQuery(that).parent().find("input").val(data.id);
-                    jQuery(that).parent().find(".treePickerTitle").text(data.name).show();
-                    jQuery(that).parent().find(".clear").show();
+                    var p = jQuery(that).parent();
+                    p.find(".buttons").show();
+
+                    p.find("input").val(data.id);
+                    p.find(".treePickerTitle").text(data.name).show();
+                    p.find(".clear").show();
                 }
             });