From 78f1ee3a94662414745c6b18554eaf7a15282cd2 Mon Sep 17 00:00:00 2001
From: SiboVG <sibo.vangool@hotmail.com>
Date: Thu, 29 Jun 2023 19:25:03 +0200
Subject: [PATCH] Better exception handling

---
 core/src/net/sf/openrocket/rocketcomponent/Rocket.java | 3 +++
 1 file changed, 3 insertions(+)

diff --git a/core/src/net/sf/openrocket/rocketcomponent/Rocket.java b/core/src/net/sf/openrocket/rocketcomponent/Rocket.java
index 513f16391..413e99b23 100644
--- a/core/src/net/sf/openrocket/rocketcomponent/Rocket.java
+++ b/core/src/net/sf/openrocket/rocketcomponent/Rocket.java
@@ -375,6 +375,9 @@ public class Rocket extends ComponentAssembly {
 		copyRocket.stageMap = new ConcurrentHashMap<>();
 		for( Map.Entry<Integer,AxialStage> entry : this.stageMap.entrySet()){
 			final AxialStage stage = (AxialStage)copyRocket.findComponent(entry.getValue().getID());
+			if (stage == null) {
+				throw new IllegalStateException("Stage not found in copy");
+			}
 			copyRocket.stageMap.put(entry.getKey(), stage);
 		}