diff --git a/internal/osbuild2/ostree_preptree_stage.go b/internal/osbuild2/ostree_preptree_stage.go
index 6e0ba91d5..4ae592756 100644
--- a/internal/osbuild2/ostree_preptree_stage.go
+++ b/internal/osbuild2/ostree_preptree_stage.go
@@ -1,7 +1,16 @@
 package osbuild2
 
+// Transforms the tree to an ostree layout
+
 type OSTreePrepTreeStageOptions struct {
+	// Array of group names to still keep in /etc/group
 	EtcGroupMembers []string `json:"etc_group_members,omitempty"`
+
+	// Array of arguments passed to dracut
+	InitramfsArgs []string `json:"initramfs-args,omitempty"`
+
+	// Create a regular directory for /tmp
+	TmpIsDir *bool `json:"tmp-is-dir,omitempty"`
 }
 
 func (OSTreePrepTreeStageOptions) isStageOptions() {}