diff --git a/prototype/UI/index.html b/prototype/UI/index.html
index b2f8a2dab..879470a8f 100644
--- a/prototype/UI/index.html
+++ b/prototype/UI/index.html
@@ -219,14 +219,14 @@
     <script type="text/javascript">
     loadJSON('diff.json', function (json) {
         var diff = diffFromJSON(json.diff);
-        var mapped = diff.map(function(value) {
-            return {
-                before: value.before != null ? termToDOM(value.before, "before", json["before"]) : document.createTextNode(""),
-                after: value.after != null ? termToDOM(value.after, "after", json["after"]) : document.createTextNode(""),
-            };
+        var before = diff.map(function(value) {
+            return value.before != null ? termToDOM(value.before, "before", json["before"]) : null;
         });
-        document.getElementById("before").appendChild(toDOM(mapped, "before", json["before"]));
-        document.getElementById("after").appendChild(toDOM(mapped, "after", json["after"]));
+        var after = diff.map(function(value) {
+            return value.after != null ? termToDOM(value.after, "after", json["after"]) : null;
+        });
+        document.getElementById("before").appendChild(diffToDOM(before, "before", json["before"]));
+        document.getElementById("after").appendChild(diffToDOM(after, "after", json["after"]));
     });
     </script>
   </body>