From 022cddaa4050a0d47a80a2681830a85e9926c35f Mon Sep 17 00:00:00 2001
From: Light_Quanta <zch22248@foxmail.com>
Date: Sat, 10 May 2025 17:56:29 +0800
Subject: [PATCH] =?UTF-8?q?=E6=AD=A3=E7=A1=AE=E5=BA=8F=E5=88=97=E5=8C=96En?=
 =?UTF-8?q?ergyStorage?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 .../com/atsuishio/superbwarfare/entity/DPSGeneratorEntity.java  | 2 ++
 1 file changed, 2 insertions(+)

diff --git a/src/main/java/com/atsuishio/superbwarfare/entity/DPSGeneratorEntity.java b/src/main/java/com/atsuishio/superbwarfare/entity/DPSGeneratorEntity.java
index 3fd2a6d4b..898e8a899 100644
--- a/src/main/java/com/atsuishio/superbwarfare/entity/DPSGeneratorEntity.java
+++ b/src/main/java/com/atsuishio/superbwarfare/entity/DPSGeneratorEntity.java
@@ -109,6 +109,8 @@ public class DPSGeneratorEntity extends LivingEntity implements GeoEntity {
         if (entityCap == null) return;
 
         ((SyncedEntityEnergyStorage) entityCap).setEnergy(compound.getInt("Energy"));
+        ((SyncedEntityEnergyStorage) entityCap).setCapacity(this.getMaxEnergy());
+        ((SyncedEntityEnergyStorage) entityCap).setMaxExtract(this.getMaxTransfer());
     }
 
     @Override