diff --git a/internal/osbuild2/tar_stage.go b/internal/osbuild2/tar_stage.go
index f8c87963b..995de834c 100644
--- a/internal/osbuild2/tar_stage.go
+++ b/internal/osbuild2/tar_stage.go
@@ -16,11 +16,29 @@ type TarStageOptions struct {
 
 func (TarStageOptions) isStageOptions() {}
 
+type TarStageInput struct {
+	inputCommon
+	References TarStageReferences `json:"references"`
+}
+
+func (TarStageInput) isStageInput() {}
+
+type TarStageInputs struct {
+	Tree *TarStageInput `json:"tree"`
+}
+
+func (TarStageInputs) isStageInputs() {}
+
+type TarStageReferences []string
+
+func (TarStageReferences) isReferences() {}
+
 // Assembles a tree into a tar archive. Compression is determined by the suffix
 // (i.e., --auto-compress is used).
-func NewTarAssembler(options *TarStageOptions) *Stage {
+func NewTarStage(options *TarStageOptions, inputs *TarStageInputs) *Stage {
 	return &Stage{
 		Type:    "org.osbuild.tar",
 		Options: options,
+		Inputs:  inputs,
 	}
 }