From d44a91d10bb6dc4b5d3be3629b7f0841a91636a9 Mon Sep 17 00:00:00 2001
From: Lawrence Forman <lawrence@0xproject.com>
Date: Mon, 8 Jun 2020 15:37:51 -0400
Subject: [PATCH] `@0x/contracts-zero-ex`: Address review feedback.

---
 .../contracts/src/transformers/AffiliateFeeTransformer.sol | 7 ++++---
 1 file changed, 4 insertions(+), 3 deletions(-)

diff --git a/contracts/zero-ex/contracts/src/transformers/AffiliateFeeTransformer.sol b/contracts/zero-ex/contracts/src/transformers/AffiliateFeeTransformer.sol
index 03bbafaeae..17025ef0b2 100644
--- a/contracts/zero-ex/contracts/src/transformers/AffiliateFeeTransformer.sol
+++ b/contracts/zero-ex/contracts/src/transformers/AffiliateFeeTransformer.sol
@@ -73,9 +73,10 @@ contract AffiliateFeeTransformer is
 
         // Transfer tokens to recipients.
         for (uint256 i = 0; i < fees.length; ++i) {
-            uint256 amount = fees[i].amount == MAX_UINT256
-                ? LibERC20Transformer.getTokenBalanceOf(fees[i].token, address(this))
-                : fees[i].amount;
+            uint256 amount = fees[i].amount;
+            if (amount == MAX_UINT256) {
+                amount = LibERC20Transformer.getTokenBalanceOf(fees[i].token, address(this));
+            }
             if (amount != 0) {
                 fees[i].token.transformerTransfer(fees[i].recipient, amount);
             }