1 files changed, 13 insertions, 2 deletions
diff --git a/docs/spec/domSpec.js b/docs/spec/domSpec.js
index af1a8faa..58063789 100644
--- a/docs/spec/domSpec.js
+++ b/docs/spec/domSpec.js
@@ -5,13 +5,24 @@ describe('dom', function(){
     it('should render code, live, test', function(){
       var dom = new DOM();
       dom.example('desc', 'src', 'scenario');
-      expect(dom.toString()).toEqual('<h1>Example</h1>\ndesc<doc:example><doc:source>src</doc:source>\n<doc:scenario>scenario</doc:scenario>\n</doc:example>\n');
+      expect(dom.toString()).toEqual(
+          '<h1>Example</h1>\n' +
+          '<div class="example">' +
+          'desc<doc:example><doc:source>src</doc:source>\n' +
+          '<doc:scenario>scenario</doc:scenario>\n'+
+          '</doc:example>\n' +
+          '</div>\n');
     });
 
     it('should render non-live, test with description', function(){
       var dom = new DOM();
       dom.example('desc', 'src', false);
-      expect(dom.toString()).toEqual('<h1>Example</h1>\ndesc<div ng:non-bindable=""><pre class="brush: js; html-script: true;">src</pre>\n</div>\n');
+      expect(dom.toString()).toEqual('<h1>Example</h1>\n' +
+          '<div class="example">' +
+          'desc<div ng:non-bindable="">' +
+          '<pre class="brush: js; html-script: true;">src</pre>\n' +
+          '</div>\n' +
+          '</div>\n');
     });
 
     it('should render non-live, test', function(){