1 files changed, 7 insertions, 5 deletions
diff --git a/functions/repack.pp b/functions/repack.pp
index 396354c..a4b7815 100644
--- a/functions/repack.pp
+++ b/functions/repack.pp
@@ -1,7 +1,9 @@
 # Normalizes a hash or a list of hashes to a list of hashes
-function networking::repack(Variant[Hash, Array[Hash]] $data) >> Array[Hash] {
-  $data ? {
-    Hash => [$data],
-    _    => $data,
-  }
+function networking::repack(Hash[String,Variant[Hash, Array[Hash]]] $data) >> Hash[String, Array[Hash]] {
+  $data.map |$key, $body| {
+    [$key, $body ? {
+      Hash    => [$body],
+      default => $body,
+    }]
+  }.convert_to(Hash)
 }