ui : one more step closer to Assembly's sudden death

2 files changed, 5 insertions, 5 deletions

diff --git a/core/src/ch/asynk/gdx/boardgame/ui/Assembly.java b/core/src/ch/asynk/gdx/boardgame/ui/Assembly.java
index b2c2cec..c61f810 100644
--- a/core/src/ch/asynk/gdx/boardgame/ui/Assembly.java
+++ b/core/src/ch/asynk/gdx/boardgame/ui/Assembly.java
@@ -14,11 +14,6 @@ public abstract class Assembly extends Element
         this.children = new IterableSet<Element>(c);
     }
 
-    public void taintChildren()
-    {
-        children.forEach( c -> c.taint() );
-    }
-
     public Element touched()
     {
         return touched;
diff --git a/core/src/ch/asynk/gdx/boardgame/ui/Element.java b/core/src/ch/asynk/gdx/boardgame/ui/Element.java
index 5e28c1d..db1735d 100644
--- a/core/src/ch/asynk/gdx/boardgame/ui/Element.java
+++ b/core/src/ch/asynk/gdx/boardgame/ui/Element.java
@@ -100,6 +100,11 @@ public class Element implements Drawable, Paddable, Positionable, Touchable
         }
     }
 
+    public void taintChildren()
+    {
+        if (children != null)
+            children.forEach( c -> c.taint() );
+    }
 
     @Override public void draw(Batch batch)
     {