From 541e5bd6aed006384751dfe5afccaafbf040208b Mon Sep 17 00:00:00 2001
From: Jozufozu <jozsefaug@gmail.com>
Date: Tue, 11 Jan 2022 13:43:04 -0800
Subject: [PATCH] Fix crash extending belts

---
 .../content/contraptions/relays/belt/BeltTileEntity.java | 9 +++------
 1 file changed, 3 insertions(+), 6 deletions(-)

diff --git a/src/main/java/com/simibubi/create/content/contraptions/relays/belt/BeltTileEntity.java b/src/main/java/com/simibubi/create/content/contraptions/relays/belt/BeltTileEntity.java
index 78aa5d9f6..020db1492 100644
--- a/src/main/java/com/simibubi/create/content/contraptions/relays/belt/BeltTileEntity.java
+++ b/src/main/java/com/simibubi/create/content/contraptions/relays/belt/BeltTileEntity.java
@@ -234,7 +234,9 @@ public class BeltTileEntity extends KineticTileEntity {
 			beltLength = compound.getInt("Length");
 			if (prevBeltLength != beltLength) {
 				DistExecutor.unsafeRunWhenOn(Dist.CLIENT, () -> () -> {
-					lighter = null;
+					if (lighter != null) {
+						lighter.initializeLight();
+					}
 				});
 			}
 		}
@@ -593,11 +595,6 @@ public class BeltTileEntity extends KineticTileEntity {
 			GridAlignedBB beltVolume = getVolume();
 
 			if (beltVolume.intersects(changed)) {
-				if (light == null) {
-					initializeLight();
-					return;
-				}
-
 				if (type == LightLayer.BLOCK)
 					updateBlockLight();