minor fix

Author: SatyamSB

contracts/tokens/SecurityToken.sol

@@ -538,10 +538,7 @@ contract SecurityToken is ERC20, ReentrancyGuard, SecurityTokenStorage, IERC1594
     function _returnPartition(uint256 _beforeBalance, uint256 _afterBalance, uint256 _value) internal pure returns(bytes32 toPartition) {
         // return LOCKED only when the transaction `_value` should be equal to the change in the LOCKED partition
         // balance otherwise return UNLOCKED
-        if (_afterBalance.sub(_beforeBalance) == _value)
-            toPartition = LOCKED;
-        // Returning the same partition UNLOCKED
-        toPartition = UNLOCKED;
+        toPartition = _afterBalance.sub(_beforeBalance) == _value ? LOCKED : UNLOCKED; // Returning the same partition UNLOCKED
     }
 
     ///////////////////////