diff --git a/Content.Shared/PowerCell/PowerCellSystem.cs b/Content.Shared/PowerCell/PowerCellSystem.cs index a732fc0e48..c7b303392b 100644 --- a/Content.Shared/PowerCell/PowerCellSystem.cs +++ b/Content.Shared/PowerCell/PowerCellSystem.cs @@ -135,8 +135,16 @@ public sealed partial class PowerCellSystem : EntitySystem private void OnDrawRefreshChargeRate(Entity ent, ref RefreshChargeRateEvent args) { - if (ent.Comp.Enabled) - args.NewChargeRate -= ent.Comp.DrawRate; + if (!ent.Comp.Enabled) + return; + + args.NewChargeRate -= ent.Comp.DrawRate; + + // If PowerCellDraw is enabled. then it should set the charge cooldown on the self recharging battery + if (!TryGetBatteryFromSlot(ent.Owner, out var battery)) + return; + + _battery.TrySetChargeCooldown(battery.Value.Owner); } private void OnDrawStartup(Entity ent, ref ComponentStartup args)