1 files changed, 15 insertions, 4 deletions
diff --git a/test/ebook.js b/test/ebook.js
index 9b353d2..033fd04 100644
--- a/test/ebook.js
+++ b/test/ebook.js
@@ -37,12 +37,23 @@ describe('eBook generator', function () {
                 path.join(book.options.output, "index.html"),
                 { encoding: "utf-8" }
             );
+
+            // There are 2 styles (one from plugin-highlight and the new style)
             PAGE.should.be.html({
                 "link": {
-                    count: 1,
-                    attributes: {
-                        href: "./styles/print.css"
-                    }
+                    count: 2
+                }
+            });
+
+            PAGE.should.be.html({
+                "link[href='./styles/print.css']": {
+                    count: 1
+                }
+            });
+
+            PAGE.should.be.html({
+                "link[href='gitbook/plugins/gitbook-plugin-highlight/ebook.css']": {
+                    count: 1
                 }
             });
         });