diff --git a/MLEM.Ui/UiControls.cs b/MLEM.Ui/UiControls.cs
index 21d584f..4d4a714 100644
--- a/MLEM.Ui/UiControls.cs
+++ b/MLEM.Ui/UiControls.cs
@@ -166,6 +166,8 @@ namespace MLEM.Ui {
         }
 
         public Element GetSelectedElement(RootElement root) {
+            if (root == null)
+                return null;
             this.selectedElements.TryGetValue(root.Name, out var element);
             return element;
         }