diff --git a/contracts/mocks/EnumerableSetMock.sol b/contracts/mocks/EnumerableSetMock.sol
index 0f77fc1e8..69de0a4d8 100644
--- a/contracts/mocks/EnumerableSetMock.sol
+++ b/contracts/mocks/EnumerableSetMock.sol
@@ -1,12 +1,7 @@
 pragma solidity ^0.5.0;
+
 import "../utils/EnumerableSet.sol";
 
-
-/**
- * @title EnumerableSetMock
- * @dev Data structure - https://en.wikipedia.org/wiki/Set_(abstract_data_type)
- * @author Alberto Cuesta Cañada
- */
 contract EnumerableSetMock{
     using EnumerableSet for EnumerableSet.AddressSet;
 
@@ -18,45 +13,21 @@ contract EnumerableSetMock{
         set = EnumerableSet.newAddressSet();
     }
 
-    /**
-     * @dev Returns true if the value is in the set.
-     */
-    function testContains(address value)
-        public
-        view
-        returns (bool)
-    {
+    function testContains(address value) public view returns (bool) {
         return EnumerableSet.contains(set, value);
     }
 
-    /**
-     * @dev Insert an value as the new tail.
-     */
-    function testAdd(address value)
-        public
-    {
+    function testAdd(address value) public {
         bool result = EnumerableSet.add(set, value);
         emit TransactionResult(result);
     }
 
-    /**
-     * @dev Remove an value.
-     */
-    function testRemove(address remove)
-        public
-    {
+    function testRemove(address remove) public {
         bool result = EnumerableSet.remove(set, remove);
         emit TransactionResult(result);
     }
 
-    /**
-     * @dev Return an array with all values in the set, from Head to Tail.
-     */
-    function testEnumerate()
-        public
-        view
-        returns (address[] memory)
-    {
+    function testEnumerate() public view returns (address[] memory) {
         return EnumerableSet.enumerate(set);
     }
 }