pee jay bee rahvew

EmoGarbage404 · Jun 1, 2024 · c667b29 · c667b29
1 parent 224d691
commit c667b29
Showing 1 changed file with 9 additions and 4 deletions.
diff --git a/Robust.Client/UserInterface/Controls/EntityPrototypeView.cs b/Robust.Client/UserInterface/Controls/EntityPrototypeView.cs
@@ -42,12 +42,17 @@ public void SetPrototype(EntProtoId? entProto)
         }
     }
 
-    protected override void Dispose(bool disposing)
+    protected override void EnteredTree()
     {
-        base.Dispose(disposing);
+        base.EnteredTree();
 
-        if (!disposing)
-            return;
+        if (_currentPrototype != null)
+            SetPrototype(_currentPrototype);
+    }
+
+    protected override void ExitedTree()
+    {
+        base.ExitedTree();
 
         if (!EntMan.Deleted(_ourEntity))
             EntMan.QueueDeleteEntity(_ourEntity);