From 064aefb04c3b7d62550a45d28c2806d6ee208f51 Mon Sep 17 00:00:00 2001
From: Noah Zinsmeister <noahwz@gmail.com>
Date: Fri, 31 Jan 2020 11:10:06 -0500
Subject: [PATCH] fix kLast bug when fee goes on -> off -> on

---
 contracts/UniswapV2Exchange.sol | 6 ++++--
 1 file changed, 4 insertions(+), 2 deletions(-)

diff --git a/contracts/UniswapV2Exchange.sol b/contracts/UniswapV2Exchange.sol
index 39800d3..b1a17ad 100644
--- a/contracts/UniswapV2Exchange.sol
+++ b/contracts/UniswapV2Exchange.sol
@@ -79,9 +79,9 @@ contract UniswapV2Exchange is IUniswapV2Exchange, UniswapV2ERC20 {
     function _mintFee(uint112 _reserve0, uint112 _reserve1) private returns (bool feeOn) {
         address feeTo = IUniswapV2Factory(factory).feeTo();
         feeOn = feeTo != address(0);
+        uint _kLast = kLast; // gas savings
         if (feeOn) {
-            uint _kLast = kLast; // gas savings
-            if (_kLast > 0) {
+            if (_kLast != 0) {
                 uint rootK = Math.sqrt(uint(_reserve0).mul(_reserve1));
                 uint rootKLast = Math.sqrt(_kLast);
                 if (rootK > rootKLast) {
@@ -91,6 +91,8 @@ contract UniswapV2Exchange is IUniswapV2Exchange, UniswapV2ERC20 {
                     if (liquidity > 0) _mint(feeTo, liquidity);
                 }
             }
+        } else if (_kLast != 0) {
+            kLast = 0;
         }
     }