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