From 3902569cbd34352da244f1bf44612fbacde1e284 Mon Sep 17 00:00:00 2001
From: simibubi <31564874+simibubi@users.noreply.github.com>
Date: Fri, 10 Jul 2020 12:49:24 +0200
Subject: [PATCH] Fix another kinetic loop when removing shafts from a belt
 segment

---
 .../create/content/contraptions/relays/belt/BeltBlock.java | 7 +------
 1 file changed, 1 insertion(+), 6 deletions(-)

diff --git a/src/main/java/com/simibubi/create/content/contraptions/relays/belt/BeltBlock.java b/src/main/java/com/simibubi/create/content/contraptions/relays/belt/BeltBlock.java
index a62b2d6db..8116171dc 100644
--- a/src/main/java/com/simibubi/create/content/contraptions/relays/belt/BeltBlock.java
+++ b/src/main/java/com/simibubi/create/content/contraptions/relays/belt/BeltBlock.java
@@ -299,12 +299,7 @@ public class BeltBlock extends HorizontalKineticBlock implements ITE<BeltTileEnt
 		if (state.get(PART) == BeltPart.PULLEY) {
 			if (world.isRemote)
 				return ActionResultType.SUCCESS;
-			world.setBlockState(context.getPos(), state.with(PART, BeltPart.MIDDLE), 2);
-			BeltTileEntity belt = BeltHelper.getSegmentTE(world, context.getPos());
-			if (belt != null) {
-				belt.detachKinetics();
-				belt.attachKinetics();
-			}
+			KineticTileEntity.switchToBlockState(world, pos, state.with(PART, BeltPart.MIDDLE));
 			if (player != null && !player.isCreative())
 				player.inventory.placeItemBackInInventory(world, AllBlocks.SHAFT.asStack());
 			return ActionResultType.SUCCESS;