diff --git a/js/id/ui/confirm.js b/js/id/ui/confirm.js
index d70612b87..6ce9b71ee 100644
--- a/js/id/ui/confirm.js
+++ b/js/id/ui/confirm.js
@@ -15,12 +15,17 @@ iD.ui.confirm = function(selection) {
     var buttonwrap = section.append('div')
         .attr('class', 'modal-section buttons cf');
 
-    buttonwrap.append('button')
-        .attr('class', 'col2 action')
-        .on('click.confirm', function() {
-            modal.remove();
-        })
-        .text(t('confirm.okay'));
+    modal.okButton = function() {
+        buttonwrap
+            .append('button')
+            .attr('class', 'col2 action')
+            .on('click.confirm', function() {
+                modal.remove();
+            })
+            .text(t('confirm.okay'));
+
+        return modal;
+    };
 
     return modal;
 };