diff --git a/src/fast-transfer-lib/java/badasintended/megane/impl/FastTransferLib.java b/src/fast-transfer-lib/java/badasintended/megane/impl/FastTransferLib.java
index a905236..ee7673e 100644
--- a/src/fast-transfer-lib/java/badasintended/megane/impl/FastTransferLib.java
+++ b/src/fast-transfer-lib/java/badasintended/megane/impl/FastTransferLib.java
@@ -4,16 +4,14 @@
 import badasintended.megane.api.provider.EnergyProvider;
 import badasintended.megane.api.registry.MeganeRegistrar;
 import dev.technici4n.fasttransferlib.api.energy.EnergyIo;
-import net.minecraft.block.entity.BlockEntity;
 
 public class FastTransferLib implements MeganeModule {
 
     @Override
     public void register(MeganeRegistrar registrar) {
-        registrar.energy(1100, BlockEntity.class, EnergyProvider.of(
-            t -> t instanceof EnergyIo,
-            t -> ((EnergyIo) t).getEnergy(),
-            t -> ((EnergyIo) t).getEnergyCapacity()
+        registrar.energy(1100, EnergyIo.class, EnergyProvider.of(
+            EnergyIo::getEnergy,
+            EnergyIo::getEnergyCapacity
         ));
     }
 
diff --git a/src/team-reborn-energy/java/badasintended/megane/impl/TeamRebornEnergy.java b/src/team-reborn-energy/java/badasintended/megane/impl/TeamRebornEnergy.java
index 9eb5bee..7d5f2bf 100644
--- a/src/team-reborn-energy/java/badasintended/megane/impl/TeamRebornEnergy.java
+++ b/src/team-reborn-energy/java/badasintended/megane/impl/TeamRebornEnergy.java
@@ -11,7 +11,7 @@ public class TeamRebornEnergy implements MeganeModule {
 
     @Override
     public void register(MeganeRegistrar registrar) {
-        registrar.energy(1100, BlockEntity.class, new EnergyProvider<BlockEntity>() {
+        registrar.energy(1200, BlockEntity.class, new EnergyProvider<BlockEntity>() {
             EnergyHandler handler;
 
             @Override