From 6a8478b5753493d50ab55622dabe94ed5666d009 Mon Sep 17 00:00:00 2001
From: GenZmeY <genzmey@gmail.com>
Date: Sat, 27 Mar 2021 18:23:50 +0300
Subject: [PATCH] fix endless wave end bug

---
 MskGs/Classes/MskGs_Endless.uc | 24 ++++++++++++++++++++++++
 1 file changed, 24 insertions(+)

diff --git a/MskGs/Classes/MskGs_Endless.uc b/MskGs/Classes/MskGs_Endless.uc
index b75fe55..a682072 100644
--- a/MskGs/Classes/MskGs_Endless.uc
+++ b/MskGs/Classes/MskGs_Endless.uc
@@ -111,6 +111,30 @@ function UpdateGameSettings()
 		}
 	}
 }
+	
+function WaveEnded(EWaveEndCondition WinCondition)
+{
+	if(!bWaveStarted && !MyKFGRI.bTraderIsOpen)
+		return;
+
+	WaveMax = WaveNum + 2;
+	MyKFGRI.WaveMax = WaveMax;
+
+	if(KFGameReplicationInfo_Endless(GameReplicationInfo).CurrentWeeklyMode != INDEX_NONE)
+	{
+		EndOutbreakRound();
+	}
+	bUseSpecialWave = false;
+
+	KFGameReplicationInfo_Endless(GameReplicationInfo).CurrentWeeklyMode = INDEX_NONE;
+	KFGameReplicationInfo_Endless(GameReplicationInfo).CurrentSpecialMode = INDEX_NONE;
+
+	HellOnEarthPlusRoundIncrement();
+
+	Super(KFGameInfo_Survival).WaveEnded(WinCondition);
+
+	TrySetNextWaveSpecial();
+}
 
 defaultproperties
 {