1 files changed, 5 insertions, 4 deletions

diff --git a/js/lib/NJUtils.js b/js/lib/NJUtils.js
index d0f547f5..90aebb84 100755
--- a/js/lib/NJUtils.js
+++ b/js/lib/NJUtils.js
@@ -19,16 +19,17 @@ exports.NJUtils = Montage.create(Component, {
     ///// Quick "getElementById"
     $ : {
         value: function(id, doc) {
-            var _doc = doc || document;
-            return _doc.getElementById(id);
+            doc = doc || document;
+            return doc.getElementById(id);
         }
     },
     
     ///// Quick "getElementsByClassName" which also returns as an Array
     ///// Can return as NodeList by passing true as second argument
     $$ : {
-        value: function(className, asNodeList) {
-            var list = document.getElementsByClassName(className);
+        value: function(className, asNodeList, doc) {
+            doc = doc || document;
+            var list = doc.getElementsByClassName(className);
             return (asNodeList) ? list : this.toArray(list);
         }
     },