further attempts

space-wizards · Sep 21, 2024 · d033cda · d033cda
1 parent 1120ffe
commit d033cda
Showing 1 changed file with 5 additions and 5 deletions.
diff --git a/Content.Shared/Silicons/StationAi/SharedStationAiSystem.cs b/Content.Shared/Silicons/StationAi/SharedStationAiSystem.cs
@@ -356,11 +356,11 @@ private void OnAiRemove(Entity<StationAiCoreComponent> ent, ref EntRemovedFromCo
 
     private void OnAiBrainRemoved(Entity<StationAiHeldComponent> ent, ref EntGotRemovedFromContainerMessage args)
     {
-        if(!TryComp(ent.Owner, out ActionsComponent? targetComponent)) 
-            return;
-        targetComponent.Actions.Clear();
-
-        Dirty(targetComponent.Owner, targetComponent);
+        if (TryComp(args.Entity, out ActionsComponent? comp))
+        {
+            comp.Actions.Clear();
+            Dirty(args.Entity, comp);
+        }
     }
 
     private void UpdateAppearance(Entity<StationAiHolderComponent?> entity)