Fix MR

PolymathNetwork · May 18, 2018 · 33088ee · 33088ee
1 parent da7d7dd
commit 33088ee
Showing 1 changed file with 9 additions and 8 deletions.
diff --git a/contracts/ModuleRegistry.sol b/contracts/ModuleRegistry.sol
@@ -31,14 +31,15 @@ contract ModuleRegistry is IModuleRegistry, Ownable {
     * @param _moduleFactory is the address of the relevant module factory
     */
     function useModule(address _moduleFactory) external {
-        //msg.sender must be a security token - below will throw if not
-        ISecurityTokenRegistry(securityTokenRegistry).getSecurityTokenData(msg.sender);
-        require(registry[_moduleFactory] != 0, "ModuleFactory type should not be 0");
-        //To use a module, either it must be verified, or owned by the ST owner
-        require(verified[_moduleFactory]||(IModuleFactory(_moduleFactory).owner() == ISecurityToken(msg.sender).owner()),
-          "Module factory is not verified as well as not called by the owner");
-        reputation[_moduleFactory].push(msg.sender);
-        emit LogModuleUsed (_moduleFactory, msg.sender);
+        //If caller is a registered security token, then register module usage
+        if (ISecurityTokenRegistry(securityTokenRegistry).isSecurityToken(msg.sender)) {
+            require(registry[_moduleFactory] != 0, "ModuleFactory type should not be 0");
+            //To use a module, either it must be verified, or owned by the ST owner
+            require(verified[_moduleFactory]||(IModuleFactory(_moduleFactory).owner() == ISecurityToken(msg.sender).owner()),
+              "Module factory is not verified as well as not called by the owner");
+            reputation[_moduleFactory].push(msg.sender);
+            emit LogModuleUsed (_moduleFactory, msg.sender);
+        }
     }
 
     /**