diff --git a/test/utils/structs/EnumerableMap.behavior.js b/test/utils/structs/EnumerableMap.behavior.js index c80eec934..cf8de2e9d 100644 --- a/test/utils/structs/EnumerableMap.behavior.js +++ b/test/utils/structs/EnumerableMap.behavior.js @@ -172,8 +172,12 @@ function shouldBehaveLikeMap() { it('missing value', async function () { await expect(this.methods.get(this.keyB)) - .to.be.revertedWithCustomError(this.mock, 'EnumerableMapNonexistentKey') - .withArgs(ethers.AbiCoder.defaultAbiCoder().encode([this.keyType], [this.keyB])); + .to.be.revertedWithCustomError(this.mock, this.error ?? 'EnumerableMapNonexistentKey') + .withArgs( + this.key?.memory || this.value?.memory + ? this.keyB + : ethers.AbiCoder.defaultAbiCoder().encode([this.keyType], [this.keyB]), + ); }); });