Fix variant not saving to NBT data

seaneoo · Nov 12, 2024 · 8cb0028 · 8cb0028
1 parent c349bbb
commit 8cb0028
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/src/main/kotlin/dev/seano/sgp/entity/GuineaPigEntity.kt b/src/main/kotlin/dev/seano/sgp/entity/GuineaPigEntity.kt
@@ -86,12 +86,12 @@ class GuineaPigEntity(entityType: EntityType<out TameableEntity>?, world: World?
 
 	override fun readCustomDataFromNbt(nbt: NbtCompound) {
 		super.readCustomDataFromNbt(nbt)
-		getVariant()?.let { nbt.putInt("Variant", it.id) }
+		GuineaPigVariant.byIndex(nbt.getInt("Variant"))?.let { setVariant(it) }
 	}
 
 	override fun writeCustomDataToNbt(nbt: NbtCompound) {
 		super.writeCustomDataToNbt(nbt)
-		GuineaPigVariant.byIndex(nbt.getInt("Variant"))?.let { setVariant(it) }
+		getVariant()?.let { nbt.putInt("Variant", it.id) }
 	}
 
 	override fun initialize(