From 37d686adb3b495e0800e5f790f9f1b9f72a181cf Mon Sep 17 00:00:00 2001
From: kruland2607 <kruland2607@users.noreply.github.com>
Date: Tue, 28 Oct 2014 20:47:48 -0500
Subject: [PATCH] Fix reported bug.  The mass & cg override icons wouldn't
 appear immediate when a components mass was overridden.  The
 ComponentTreeModel needs to update the tree on mass changes too.

---
 .../openrocket/gui/main/componenttree/ComponentTreeModel.java   | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/swing/src/net/sf/openrocket/gui/main/componenttree/ComponentTreeModel.java b/swing/src/net/sf/openrocket/gui/main/componenttree/ComponentTreeModel.java
index fe141f612..c80b73561 100644
--- a/swing/src/net/sf/openrocket/gui/main/componenttree/ComponentTreeModel.java
+++ b/swing/src/net/sf/openrocket/gui/main/componenttree/ComponentTreeModel.java
@@ -138,7 +138,7 @@ public class ComponentTreeModel implements TreeModel, ComponentChangeListener {
 	
 	@Override
 	public void componentChanged(ComponentChangeEvent e) {
-		if (e.isTreeChange() || e.isUndoChange()) {
+		if (e.isTreeChange() || e.isUndoChange() || e.isMassChange()) {
 			// Tree must be fully updated also in case of an undo change 
 			fireTreeStructureChanged(e.getSource());
 			if (e.isTreeChange() && e.isUndoChange()) {