diff --git a/Content.Server/Light/EntitySystems/HandheldLightSystem.cs b/Content.Server/Light/EntitySystems/HandheldLightSystem.cs index 90b9a812c7..813f8c407b 100644 --- a/Content.Server/Light/EntitySystems/HandheldLightSystem.cs +++ b/Content.Server/Light/EntitySystems/HandheldLightSystem.cs @@ -247,13 +247,16 @@ namespace Content.Server.Light.EntitySystems public void TryUpdate(Entity uid, float frameTime) { var component = uid.Comp; - if (!_powerCell.TryGetBatteryFromSlot(uid, out var battery) && + if (!_powerCell.TryGetBatteryFromSlot(uid, out var batteryUid, out var battery, null) && !TryComp(uid, out battery)) { TurnOff(uid, false); return; } + if (batteryUid == null) + return; + var appearanceComponent = EntityManager.GetComponentOrNull(uid); var fraction = battery.CurrentCharge / battery.MaxCharge; @@ -270,7 +273,7 @@ namespace Content.Server.Light.EntitySystems _appearance.SetData(uid, HandheldLightVisuals.Power, HandheldLightPowerStates.Dying, appearanceComponent); } - if (component.Activated && !_battery.TryUseCharge(uid, component.Wattage * frameTime, battery)) + if (component.Activated && !_battery.TryUseCharge(batteryUid.Value, component.Wattage * frameTime, battery)) TurnOff(uid, false); UpdateLevel(uid); diff --git a/Content.Server/Radio/EntitySystems/JammerSystem.cs b/Content.Server/Radio/EntitySystems/JammerSystem.cs index e6a8701f20..dc6f476416 100644 --- a/Content.Server/Radio/EntitySystems/JammerSystem.cs +++ b/Content.Server/Radio/EntitySystems/JammerSystem.cs @@ -30,8 +30,8 @@ public sealed class JammerSystem : EntitySystem var query = EntityQueryEnumerator(); while (query.MoveNext(out var uid, out var _, out var jam)) { - if (_powerCell.TryGetBatteryFromSlot(uid, out var battery) && - !_battery.TryUseCharge(uid, jam.Wattage * frameTime, battery)) + if (_powerCell.TryGetBatteryFromSlot(uid, out var batteryUid, out var battery) && + !_battery.TryUseCharge(batteryUid.Value, jam.Wattage * frameTime, battery)) { RemComp(uid); }