diff --git a/public/assets/js/app.js b/public/assets/js/app.js
index d444c12..ade8a62 100644
--- a/public/assets/js/app.js
+++ b/public/assets/js/app.js
@@ -184,7 +184,7 @@
 		};
 
 		$scope.refreshHighlighting = function() {
-			var sourceCodes = document.querySelectorAll('main .file .code.source');
+			var sourceCodes = $window.document.querySelectorAll('main .file .code.source');
 
 			for (var i = 0; i < sourceCodes.length; i++) {
 				var sourceCode = sourceCodes[i];
@@ -260,7 +260,7 @@
 		$scope.downloadZip = function() {
 			var zip = new JSZip();
 
-			var sourceCodes = document.querySelectorAll('main .file .code.source');
+			var sourceCodes = $window.document.querySelectorAll('main .file .code.source');
 
 			for (var i = 0; i < sourceCodes.length; i++) {
 				var sourceCode = sourceCodes[i];