Fix Fabricators-of-Create#1552

Creators-of-Create · Nov 4, 2024 · f3bb31e · f3bb31e
1 parent d39f899
commit f3bb31e
Showing 1 changed file with 4 additions and 1 deletion.
diff --git a/...main/java/com/simibubi/create/content/kinetics/simpleRelays/AbstractSimpleShaftBlock.java b/...main/java/com/simibubi/create/content/kinetics/simpleRelays/AbstractSimpleShaftBlock.java
@@ -37,7 +37,10 @@ public PushReaction getPistonPushReaction(BlockState state) {
 
 	@Override
 	public void onRemove(BlockState state, Level world, BlockPos pos, BlockState newState, boolean isMoving) {
-		if (state != newState && !isMoving)
+		boolean wasWaterLogged = state.hasProperty(WATERLOGGED) &&
+				newState.hasProperty(WATERLOGGED) &&
+				(state.getValue(WATERLOGGED) != newState.getValue(WATERLOGGED));
+		if (state != newState && !isMoving && !wasWaterLogged)
 			removeBracket(world, pos, true).ifPresent(stack -> Block.popResource(world, pos, stack));
 		super.onRemove(state, world, pos, newState, isMoving);
 	}