From d0134a37018b8058f78ae8201f782aa2d938e263 Mon Sep 17 00:00:00 2001
From: Cameron Taylor <cameron.taylor.ninja@gmail.com>
Date: Thu, 3 Feb 2022 10:39:07 -0500
Subject: [PATCH] song difficulty swapper

---
 source/charting/ChartingState.hx | 5 +++++
 1 file changed, 5 insertions(+)

diff --git a/source/charting/ChartingState.hx b/source/charting/ChartingState.hx
index c52326083..8d6f93ef8 100644
--- a/source/charting/ChartingState.hx
+++ b/source/charting/ChartingState.hx
@@ -254,6 +254,11 @@ class ChartingState extends MusicBeatState
 		{
 			SongLoad.curDiff = _song.difficulties[Std.parseInt(diff)];
 			SongLoad.checkAndCreateNotemap(SongLoad.curDiff);
+
+			while (SongLoad.getSong()[curSection] == null)
+				addSection();
+
+			updateGrid();
 		});
 		difficultyDropDown.selectedLabel = SongLoad.curDiff;