Avoid item render leaks

IrisShaders · Sep 2, 2023 · b00491a · b00491a
1 parent 941e71c
commit b00491a
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 1 deletion.
diff --git a/src/main/java/net/coderbot/iris/mixin/entity_render_context/MixinEntityRenderDispatcher.java b/src/main/java/net/coderbot/iris/mixin/entity_render_context/MixinEntityRenderDispatcher.java
@@ -68,5 +68,6 @@ public class MixinEntityRenderDispatcher {
 									  PoseStack poseStack, MultiBufferSource bufferSource, int light,
 									  CallbackInfo ci) {
 		CapturedRenderingState.INSTANCE.setCurrentEntity(0);
+		CapturedRenderingState.INSTANCE.setCurrentRenderedItem(0);
 	}
 }
diff --git a/src/main/java/net/coderbot/iris/mixin/entity_render_context/MixinItemRenderer.java b/src/main/java/net/coderbot/iris/mixin/entity_render_context/MixinItemRenderer.java
@@ -45,7 +45,7 @@ private void changeId(ItemStack pItemRenderer0, ItemTransforms.TransformType pIt
 		}
 	}
 
-	@Inject(method = "render", at = @At(value = "TAIL"))
+	@Inject(method = "render", at = @At(value = "RETURN"))
 	private void changeId2(CallbackInfo ci) {
 		CapturedRenderingState.INSTANCE.setCurrentRenderedItem(0);
 		CapturedRenderingState.INSTANCE.setCurrentBlockEntity(previousBeValue);