From e5261ef454634fa77ddee874e477911514062b49 Mon Sep 17 00:00:00 2001
From: DBotThePony <dbotthepony@yandex.ru>
Date: Fri, 2 Dec 2022 01:20:49 +0700
Subject: [PATCH] also this

---
 .../otm/block/entity/BatteryBankBlockEntity.kt  | 17 +++++++++++------
 1 file changed, 11 insertions(+), 6 deletions(-)

diff --git a/src/main/kotlin/ru/dbotthepony/mc/otm/block/entity/BatteryBankBlockEntity.kt b/src/main/kotlin/ru/dbotthepony/mc/otm/block/entity/BatteryBankBlockEntity.kt
index 6274fc3f9..0247fdf29 100644
--- a/src/main/kotlin/ru/dbotthepony/mc/otm/block/entity/BatteryBankBlockEntity.kt
+++ b/src/main/kotlin/ru/dbotthepony/mc/otm/block/entity/BatteryBankBlockEntity.kt
@@ -244,9 +244,12 @@ class BatteryBankBlockEntity(p_155229_: BlockPos, p_155230_: BlockState) : Matte
 		resolverEnergyExtractor.invalidate()
 		resolverEnergy.invalidate()
 		resolverItemHandler.invalidate()
-		mekanismResolverEnergyReceiver.invalidate()
-		mekanismResolverEnergyExtractor.invalidate()
-		mekanismResolverEnergy.invalidate()
+
+		if (isMekanismLoaded) {
+			mekanismResolverEnergyReceiver.invalidate()
+			mekanismResolverEnergyExtractor.invalidate()
+			mekanismResolverEnergy.invalidate()
+		}
 	}
 
 	override fun reviveCaps() {
@@ -257,9 +260,11 @@ class BatteryBankBlockEntity(p_155229_: BlockPos, p_155230_: BlockState) : Matte
 		resolverEnergy = LazyOptional.of { energy }
 		resolverItemHandler = LazyOptional.of { itemHandler }
 
-		mekanismResolverEnergyReceiver = LazyOptional.of { Mattery2MekanismEnergyWrapper(energyReceiver) }
-		mekanismResolverEnergyExtractor = LazyOptional.of { Mattery2MekanismEnergyWrapper(energyExtractor) }
-		mekanismResolverEnergy = LazyOptional.of { Mattery2MekanismEnergyWrapper(energy) }
+		if (isMekanismLoaded) {
+			mekanismResolverEnergyReceiver = LazyOptional.of { Mattery2MekanismEnergyWrapper(energyReceiver) }
+			mekanismResolverEnergyExtractor = LazyOptional.of { Mattery2MekanismEnergyWrapper(energyExtractor) }
+			mekanismResolverEnergy = LazyOptional.of { Mattery2MekanismEnergyWrapper(energy) }
+		}
 	}
 
 	override fun setLevel(p_155231_: Level) {