From 91b93debec2fc47e8b271b253f6db725555220bb Mon Sep 17 00:00:00 2001
From: SiboVG <sibo.vangool@hotmail.com>
Date: Sat, 19 Feb 2022 13:01:48 +0100
Subject: [PATCH] [fixes #358] Fix Booster deletion

---
 .../sf/openrocket/rocketcomponent/FlightConfiguration.java   | 2 +-
 swing/src/net/sf/openrocket/gui/main/RocketActions.java      | 5 +++--
 2 files changed, 4 insertions(+), 3 deletions(-)

diff --git a/core/src/net/sf/openrocket/rocketcomponent/FlightConfiguration.java b/core/src/net/sf/openrocket/rocketcomponent/FlightConfiguration.java
index f7a3e2d02..49377e579 100644
--- a/core/src/net/sf/openrocket/rocketcomponent/FlightConfiguration.java
+++ b/core/src/net/sf/openrocket/rocketcomponent/FlightConfiguration.java
@@ -208,7 +208,7 @@ public class FlightConfiguration implements FlightConfigurableParameter<FlightCo
 			return true;
 		}
 		
-		return stages.get(stageNumber).active;
+		return stages.get(stageNumber) != null && stages.get(stageNumber).active;
 	}
 
 	public Collection<RocketComponent> getAllComponents() {
diff --git a/swing/src/net/sf/openrocket/gui/main/RocketActions.java b/swing/src/net/sf/openrocket/gui/main/RocketActions.java
index ae9920d54..d945d653d 100644
--- a/swing/src/net/sf/openrocket/gui/main/RocketActions.java
+++ b/swing/src/net/sf/openrocket/gui/main/RocketActions.java
@@ -25,6 +25,7 @@ import net.sf.openrocket.gui.util.Icons;
 import net.sf.openrocket.l10n.Translator;
 import net.sf.openrocket.rocketcomponent.ComponentChangeEvent;
 import net.sf.openrocket.rocketcomponent.ComponentChangeListener;
+import net.sf.openrocket.rocketcomponent.ParallelStage;
 import net.sf.openrocket.rocketcomponent.Rocket;
 import net.sf.openrocket.rocketcomponent.RocketComponent;
 import net.sf.openrocket.rocketcomponent.AxialStage;
@@ -176,8 +177,8 @@ public class RocketActions {
 		if (c instanceof Rocket)
 			return false;
 
-		// Cannot remove last stage
-		if ((c instanceof AxialStage) && (c.getParent().getChildCount() == 1)) {
+		// Cannot remove last stage, except from Boosters
+		if ((c instanceof AxialStage) && !(c instanceof ParallelStage) && (c.getParent().getChildCount() == 1)) {
 			return false;
 		}