diff --git a/prototype/UI/index.html b/prototype/UI/index.html
index f29d393b5..4ae471054 100644
--- a/prototype/UI/index.html
+++ b/prototype/UI/index.html
@@ -110,14 +110,15 @@
         }
 
         if (model instanceof Diff) {
-            element = document.createElement("div");
-            element.classList.add("diff");
             if (model.pure != null) {
+                element = document.createElement("div");
+                element.classList.add("diff");
                 element.classList.add("pure");
-                element.appendChild(toDOM(model.pure));
+                element.appendChild(toDOM(model.pure[stateName], stateName, source));
             } else if(model.roll != null) {
+                element = rangeAndSyntaxToDOM(model.ranges[stateName], mapped, source, function(child) { return toDOM(child, stateName, source); });
+                element.classList.add("diff");
                 element.classList.add("roll");
-                element.appendChild(toDOM(model.roll));
             }
         }