diff --git a/MLEM.Ui/UiSystem.cs b/MLEM.Ui/UiSystem.cs
index dfcc70d..c516c3b 100644
--- a/MLEM.Ui/UiSystem.cs
+++ b/MLEM.Ui/UiSystem.cs
@@ -134,6 +134,7 @@ namespace MLEM.Ui {
             root.Element.AndChildren(e => {
                 e.Root = root;
                 e.System = this;
+                e.SetAreaDirty();
             });
             return true;
         }
@@ -147,6 +148,7 @@ namespace MLEM.Ui {
             root.Element.AndChildren(e => {
                 e.Root = null;
                 e.System = null;
+                e.SetAreaDirty();
             });
         }