tidy amplitude menu code

1 files changed, 7 insertions, 6 deletions

diff --git a/menus.c b/menus.c
index cc91bc7..91f97ee 100644
--- a/menus.c
+++ b/menus.c
@@ -1351,6 +1351,9 @@ static void Submenu_Display(int change_selection)
 	case Submenu1_amp:
 		title = title_with_channel("Amplitude",globals.Flash.ChanKey_amplitude,channel);
 
+		Submenu_max_entry=0;
+		Submenu_Structure[0]=mode_amp_normal;
+
 		if (number_of_fixed_ampl_points(channel)>0) {
 			int fixed_count;
 			Submenu_max_entry = number_of_fixed_ampl_points(channel) - 1;
@@ -1362,14 +1365,12 @@ static void Submenu_Display(int change_selection)
 		}
 
 		if (globals.Flash.ea_enabled[channel]) {
-			Submenu_max_entry=1;
-			Submenu_Structure[0]=mode_amp_normal;
-			Submenu_Structure[1]=mode_amp_ea;
+			++Submenu_max_entry;
+			Submenu_Structure[Submenu_max_entry]=mode_amp_ea;
 		}
 		if (globals.Flash.ext_amplify_enabled[channel]) {
-			Submenu_max_entry=1;
-			Submenu_Structure[0]=mode_amp_normal;
-			Submenu_Structure[1]=mode_amp_amplify;
+			++Submenu_max_entry;
+			Submenu_Structure[Submenu_max_entry]=mode_amp_amplify;
 		}
 
 		if (globals.Flash.ignore_ampl_polarity[channel]) {