Made onlyModuleOrOwner definition consisitent (#674)

* Made onlyModuleOrOwner definition consisitent * Added braces * Added braces
PolymathNetwork · May 31, 2019 · 77e68d0 · 77e68d0
1 parent f307619
commit 77e68d0
Show file tree

Hide file tree

Showing 2 changed files with 9 additions and 5 deletions.
diff --git a/contracts/ModuleRegistry.sol b/contracts/ModuleRegistry.sol
@@ -80,10 +80,13 @@ contract ModuleRegistry is IModuleRegistry, EternalStorage {
      * @notice Modifier to make a function callable only when the contract is not paused.
      */
     modifier whenNotPausedOrOwner() {
-        if (msg.sender == owner()) _;
-        else {
-            require(!isPaused(), "Already paused");
-            _;
+        _whenNotPausedOrOwner();
+        _;
+    }
+
+    function _whenNotPausedOrOwner() internal view {
+        if (msg.sender != owner()) {
+            require(!isPaused(), "Paused");
         }
     }
 

diff --git a/contracts/SecurityTokenRegistry.sol b/contracts/SecurityTokenRegistry.sol
@@ -171,8 +171,9 @@ contract SecurityTokenRegistry is EternalStorage, Proxy {
     }
 
     function _whenNotPausedOrOwner() internal view {
-        if (msg.sender != owner())
+        if (msg.sender != owner()) {
             require(!isPaused(), "Paused");
+        }
     }
 
     /**