From 67468e464b038736251f18e1dc2e368c9091a87a Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?Alberto=20Cuesta=20Ca=C3=B1ada?= <mail@albertocuesta.es>
Date: Mon, 27 Jan 2020 14:55:59 +0000
Subject: [PATCH] Simplified mock.

---
 contracts/mocks/EnumerableSetMock.sol | 8 ++++----
 1 file changed, 4 insertions(+), 4 deletions(-)

diff --git a/contracts/mocks/EnumerableSetMock.sol b/contracts/mocks/EnumerableSetMock.sol
index 407ee6c15..4ef4fa04a 100644
--- a/contracts/mocks/EnumerableSetMock.sol
+++ b/contracts/mocks/EnumerableSetMock.sol
@@ -14,20 +14,20 @@ contract EnumerableSetMock{
     }
 
     function contains(address value) public view returns (bool) {
-        return EnumerableSet.contains(set, value);
+        return set.contains(value);
     }
 
     function add(address value) public {
-        bool result = EnumerableSet.add(set, value);
+        bool result = set.add(value);
         emit TransactionResult(result);
     }
 
     function remove(address value) public {
-        bool result = EnumerableSet.remove(set, value);
+        bool result = set.remove(value);
         emit TransactionResult(result);
     }
 
     function enumerate() public view returns (address[] memory) {
-        return EnumerableSet.enumerate(set);
+        return set.enumerate();
     }
 }