diff --git a/demo/demo.js b/demo/demo.js
index f588a8f4b..a1524f5c4 100644
--- a/demo/demo.js
+++ b/demo/demo.js
@@ -21,6 +21,7 @@ var selfieCard = {
               '<div>' +
                 '<img src="' + payload.imageSrc + '"><br>' +
                 '<div>You look nice today.</div>' +
+                (env.edit ? "<div><button id='go-edit'>Take a better picture</button></div>" : "") +
                 '<div><button id="go-edit">Take a better picture</button></div>' +
               '</div>' +
             '')[0]
@@ -29,7 +30,7 @@ var selfieCard = {
         element.appendChild($('' +
           '<div>' +
             'Hello there!' +
-            '<button id="go-edit">Click here to take a picture</button>' +
+            (env.edit ? "<button id='go-edit'>Click here to take a picture</button>" : "")
           '</div>')[0]
         );
       }
@@ -174,10 +175,14 @@ var ContentKitDemo = exports.ContentKitDemo = {
     var mobiledoc = editor.serialize();
     codePaneJSON.innerHTML = this.syntaxHighlight(mobiledoc);
 
+    var cards = {
+      'simple-card': simpleCard,
+      'edit-card': cardWithEditMode,
+      'input-card': cardWithInput,
+      'selfie-card': selfieCard
+    };
     var renderer = new MobiledocDOMRenderer();
-    var rendered = renderer.render(mobiledoc, document.createElement('div'), {
-      /* cards */
-    });
+    var rendered = renderer.render(mobiledoc, document.createElement('div'), cards);
 
     $('#rendered-mobiledoc').empty();
     $('#rendered-mobiledoc')[0].appendChild(rendered);
diff --git a/package.json b/package.json
index c0ab01446..912694064 100644
--- a/package.json
+++ b/package.json
@@ -45,8 +45,8 @@
     "broccoli-test-builder": "^0.1.0",
     "content-kit-utils": "^0.2.0",
     "jquery": "^2.1.4",
-    "mobiledoc-dom-renderer": "^0.1.5",
-    "mobiledoc-html-renderer": "^0.1.2",
+    "mobiledoc-dom-renderer": "^0.1.6",
+    "mobiledoc-html-renderer": "^0.1.3",
     "testem": "^0.8.4"
   }
 }