diff --git a/js/twitter.js b/js/twitter.js
index 9229717..823ebc9 100644
--- a/js/twitter.js
+++ b/js/twitter.js
@@ -8,7 +8,7 @@
   Drupal.behaviors.twitterMediaEntity = {
     attach: function (context) {
       function _init () {
-        twttr.widgets.load(context);
+        twttr.widgets.load(_.isElement(context) ? context : null);
       }
 
       // If the tweet is being embedded in a CKEditor's iFrame the widgets