diff --git a/dev/js/spec/pluginSpec.js b/dev/js/spec/pluginSpec.js
index fa74486..62aa3c7 100644
--- a/dev/js/spec/pluginSpec.js
+++ b/dev/js/spec/pluginSpec.js
@@ -13,7 +13,8 @@
       var div = document.createElement("div");
       var id = manager.addWidget(div, 'about:blank');
       expect(id).toMatch(/^id-/);
-      expect(div.firstChild.tagName).toEqual("IFRAME");
+      expect(div.firstChild.classList.contains('widget')).toBeTruthy();
+      expect(div.firstChild.firstChild.tagName).toEqual("IFRAME");
       manager.destroy();
     });
   });