diff --git a/upgradeability_ownership/test/helpers/assertRevert.js b/upgradeability_ownership/test/helpers/assertRevert.js
index 86bfea2..7e3a1f2 100644
--- a/upgradeability_ownership/test/helpers/assertRevert.js
+++ b/upgradeability_ownership/test/helpers/assertRevert.js
@@ -1,5 +1,10 @@
-async function assertRevert(promise) {
+async function assertRevert(promise, errMsg) {
   try {
+    if (typeof errMsg !== "undefined") {
+      console.log(errMsg);
+      const errMsgFound = errMsg.search('from the contract') >= 0;
+      assert(errMsgFound, `Expected "reason provided from the contract", got ${errMsg} instead`);
+    }
     await promise;
     assert.fail('Expected revert not received');
   } catch (error) {
@@ -8,4 +13,4 @@ async function assertRevert(promise) {
   }
 }
 
-module.exports = assertRevert;
\ No newline at end of file
+module.exports = assertRevert;