[DAE-108] Syncing into the future - fix

input-output-hk · Mar 1, 2017 · 35286c9 · 35286c9
1 parent f9dfb6a
commit 35286c9
Showing 1 changed file with 4 additions and 1 deletion.
diff --git a/src/Pos/Wallet/WalletMode.hs b/src/Pos/Wallet/WalletMode.hs
@@ -245,7 +245,10 @@ downloadHeader = getContextTMVar PC.ncProgressHeader
 instance SscHelpersClass ssc =>
          MonadBlockchainInfo (RawRealMode ssc) where
     networkChainDifficulty = getContextTVar PC.ncLastKnownHeader >>= \case
-        Just lh -> return . Just $ lh ^. difficultyL
+        Just lh -> do
+            thDiff <- view difficultyL <$> topHeader @ssc
+            let lhDiff = lh ^. difficultyL
+            return . Just $ max thDiff lhDiff
         Nothing -> runMaybeT $ do
             cSlot <- flattenSlotId <$> MaybeT getCurrentSlot
             th <- lift (topHeader @ssc)