From e81cd5b68e2e1073fc81e974ce5301c75426eca5 Mon Sep 17 00:00:00 2001
From: Anuken <arnukren@gmail.com>
Date: Fri, 31 Mar 2023 22:02:57 -0400
Subject: [PATCH] Better impl of #8451

---
 core/src/mindustry/world/blocks/power/Battery.java | 8 ++++++--
 1 file changed, 6 insertions(+), 2 deletions(-)

diff --git a/core/src/mindustry/world/blocks/power/Battery.java b/core/src/mindustry/world/blocks/power/Battery.java
index 75ebcfbd5b..043061c5a7 100644
--- a/core/src/mindustry/world/blocks/power/Battery.java
+++ b/core/src/mindustry/world/blocks/power/Battery.java
@@ -11,8 +11,6 @@ import mindustry.gen.*;
 import mindustry.world.draw.*;
 import mindustry.world.meta.*;
 
-import static mindustry.Vars.*;
-
 public class Battery extends PowerDistributor{
     public DrawBlock drawer;
 
@@ -39,6 +37,10 @@ public class Battery extends PowerDistributor{
     public void init(){
         super.init();
 
+        checkDrawDefault();
+    }
+
+    void checkDrawDefault(){
         if(drawer == null){
             drawer = new DrawMulti(new DrawDefault(), new DrawPower(){{
                 emptyLightColor = Battery.this.emptyLightColor;
@@ -49,6 +51,8 @@ public class Battery extends PowerDistributor{
 
     @Override
     public void load(){
+        checkDrawDefault();
+
         super.load();
         drawer.load(this);
     }