diff --git a/dist/angular-pdfjs-viewer.js b/dist/angular-pdfjs-viewer.js
index 55f9917..6f46933 100644
--- a/dist/angular-pdfjs-viewer.js
+++ b/dist/angular-pdfjs-viewer.js
@@ -484,9 +484,9 @@
                     var pages = document.querySelectorAll('.page');
                     angular.forEach(pages, function (page) {
                         var element = angular.element(page);
-                        var pageNum = element.data('page-number');
+                        var pageNum = element.attr('data-page-number');
                         
-                        if (!element.data('loaded')) {
+                        if (!element.attr('data-loaded')) {
                             delete loaded[pageNum];
                             return;
                         }
diff --git a/src/angular-pdfjs-viewer.js b/src/angular-pdfjs-viewer.js
index 91a79d7..030e048 100644
--- a/src/angular-pdfjs-viewer.js
+++ b/src/angular-pdfjs-viewer.js
@@ -115,9 +115,9 @@
                     var pages = document.querySelectorAll('.page');
                     angular.forEach(pages, function (page) {
                         var element = angular.element(page);
-                        var pageNum = element.data('page-number');
+                        var pageNum = element.attr('data-page-number');
                         
-                        if (!element.data('loaded')) {
+                        if (!element.attr('data-loaded')) {
                             delete loaded[pageNum];
                             return;
                         }