From 191dc871b82b2eba01115057e152517ff0c6a0fd Mon Sep 17 00:00:00 2001
From: Nemric <emeric.chassagne@outlook.fr>
Date: Thu, 16 Dec 2021 18:53:12 +0000
Subject: [PATCH] update translator

---
 config/v3_4_experimental/translate/translate.go | 11 +++++++++++
 1 file changed, 11 insertions(+)

diff --git a/config/v3_4_experimental/translate/translate.go b/config/v3_4_experimental/translate/translate.go
index 5b39cae9b..3b6deb9ae 100644
--- a/config/v3_4_experimental/translate/translate.go
+++ b/config/v3_4_experimental/translate/translate.go
@@ -27,9 +27,20 @@ func translateIgnition(old old_types.Ignition) (ret types.Ignition) {
 	return
 }
 
+func translateUnit(old old_types.Unit) (ret types.Unit) {
+	tr := translate.NewTranslator()
+	tr.Translate(&old.Contents, &ret.Contents)
+	tr.Translate(&old.Dropins, &ret.Dropins)
+	tr.Translate(&old.Enabled, &ret.Enabled)
+	tr.Translate(&old.Mask, &ret.Mask)
+	tr.Translate(&old.Name, &ret.Name)
+	return
+}
+
 func Translate(old old_types.Config) (ret types.Config) {
 	tr := translate.NewTranslator()
 	tr.AddCustomTranslator(translateIgnition)
+	tr.AddCustomTranslator(translateUnit)
 	tr.Translate(&old, &ret)
 	return
 }