diff --git a/assertions/screenshotIdenticalToBaseline.js b/assertions/screenshotIdenticalToBaseline.js
index 9934c59..b0fd1b8 100644
--- a/assertions/screenshotIdenticalToBaseline.js
+++ b/assertions/screenshotIdenticalToBaseline.js
@@ -53,7 +53,10 @@ exports.assertion = function screenshotIdenticalToBaseline(
                 compareWithBaseline(this.api, screenshot, fileName, settings).then((result) => {
                     comparisonResult = result
                     done()
-                })
+                }, (reject) => {
+                    comparisonResult = reject
+                    done()
+                });
             })
             .perform((done) => {
                 callback(comparisonResult)