diff --git a/custom.js b/custom.js index f084a32..3fa472c 100644 --- a/custom.js +++ b/custom.js @@ -1,6 +1,6 @@ var fs = require('fs'), document = require('jsdom').jsdom(''), - window = document.parentWindow; + window = document.parentWindow ? document.parentWindow : document.defaultView; module.exports = function (path) { // read angular source into memory