diff --git a/contracts/OneSplitAudit.sol b/contracts/OneSplitAudit.sol
index 9a4d866..8d51b00 100644
--- a/contracts/OneSplitAudit.sol
+++ b/contracts/OneSplitAudit.sol
@@ -232,7 +232,7 @@ contract OneSplitAudit is IOneSplit, Ownable {
         uint256[] memory flagsArray = new uint256[](1);
         flagsArray[0] = flags;
 
-        swapWithReferralMulti(
+        return swapWithReferralMulti(
             tokens,
             amount,
             minReturn,
@@ -256,7 +256,7 @@ contract OneSplitAudit is IOneSplit, Ownable {
         uint256[] memory distribution,
         uint256[] memory flags
     ) public payable returns(uint256) {
-        swapWithReferralMulti(
+        return swapWithReferralMulti(
             tokens,
             amount,
             minReturn,