diff --git a/src/main/java/com/minelittlepony/unicopia/mixin/MixinItemEntity.java b/src/main/java/com/minelittlepony/unicopia/mixin/MixinItemEntity.java index c8ee373f..3ffc0a25 100644 --- a/src/main/java/com/minelittlepony/unicopia/mixin/MixinItemEntity.java +++ b/src/main/java/com/minelittlepony/unicopia/mixin/MixinItemEntity.java @@ -47,14 +47,14 @@ abstract class MixinItemEntity extends Entity implements IItemEntity { @Inject(method = "writeCustomDataToTag(Lnet/minecraft/nbt/CompoundTag;)V", at = @At("HEAD")) private void onWriteCustomDataToTag(CompoundTag tag, CallbackInfo info) { - if (tag.contains("unicopia_caster")) { - get().fromNBT(tag.getCompound("unicopia_caster")); - } + tag.put("unicopia_caster", get().toNBT()); } @Inject(method = "readCustomDataFromTag(Lnet/minecraft/nbt/CompoundTag;)V", at = @At("HEAD")) private void onReadCustomDataFromTag(CompoundTag tag, CallbackInfo info) { - tag.put("unicopia_caster", get().toNBT()); + if (tag.contains("unicopia_caster")) { + get().fromNBT(tag.getCompound("unicopia_caster")); + } } @Accessor("age")