diff --git a/internal/osbuild/ignition_stage.go b/internal/osbuild/ignition_stage.go
index 32af0b678..e21c689d3 100644
--- a/internal/osbuild/ignition_stage.go
+++ b/internal/osbuild/ignition_stage.go
@@ -18,24 +18,14 @@ func NewIgnitionStage(options *IgnitionStageOptions) *Stage {
 }
 
 type IgnitionStageInputInline struct {
-	InlineFile IgnitionStageInput `json:"inlinefile"`
+	InlineFile *FilesInput `json:"inlinefile"`
 }
 
 func (IgnitionStageInputInline) isStageInputs() {}
 
-type IgnitionStageInput struct {
-	inputCommon
-	References IgnitionStageReferences `json:"references"`
-}
-
-type IgnitionStageReferences []string
-
-func (IgnitionStageReferences) isReferences() {}
-
 func NewIgnitionInlineInput(embeddedData string) Inputs {
-	inputs := new(IgnitionStageInputInline)
-	inputs.InlineFile.Type = "org.osbuild.files"
-	inputs.InlineFile.Origin = "org.osbuild.source"
-	inputs.InlineFile.References = IgnitionStageReferences{fmt.Sprintf("sha256:%x", sha256.Sum256([]byte(embeddedData)))}
-	return inputs
+	input := NewFilesInput(NewFilesInputSourcePlainRef([]string{
+		fmt.Sprintf("%x", sha256.Sum256([]byte(embeddedData))),
+	}))
+	return &IgnitionStageInputInline{InlineFile: input}
 }