fix: Fix last commit possibly resulting in Fabric data not saving cor…

…rectly when migrated
TwelveIterationMods · Jan 11, 2025 · 75f47e1 · 75f47e1
1 parent f9cc651
commit 75f47e1
Showing 1 changed file with 6 additions and 0 deletions.
diff --git a/fabric/src/main/java/net/blay09/mods/balm/fabric/FabricBalmHooks.java b/fabric/src/main/java/net/blay09/mods/balm/fabric/FabricBalmHooks.java
@@ -59,10 +59,16 @@ public CompoundTag getPersistentData(Entity entity) {
         if (balmData.isEmpty()) {
             // If we have no data, try to import from NeoForge in case the world was migrated
             balmData = ((BalmEntity) entity).getNeoForgeBalmData();
+            if (!balmData.isEmpty()) {
+                ((BalmEntity) entity).setFabricBalmData(balmData);
+            }
         }
         if (balmData.isEmpty()) {
             // If we still have no data, try to import from Forge in case the world was migrated
             balmData = ((BalmEntity) entity).getForgeBalmData();
+            if (!balmData.isEmpty()) {
+                ((BalmEntity) entity).setFabricBalmData(balmData);
+            }
         }
         return balmData;
     }