diff --git a/Makefile b/Makefile
index 15f34f4..a9a1b80 100644
--- a/Makefile
+++ b/Makefile
@@ -2,7 +2,7 @@ arch = x86_64
 version = 39
 base_dir = $(shell pwd)
 image_repo = ghcr.io/ublue-os
-image_name = base-main
+image_name = silverblue-main
 variant = Silverblue
 
 image_repo_escaped = $(subst /,\/,$(image_repo))
diff --git a/lorax_templates/configure_upgrades.tmpl.in b/lorax_templates/configure_upgrades.tmpl.in
index 2bbb92a..26a136c 100644
--- a/lorax_templates/configure_upgrades.tmpl.in
+++ b/lorax_templates/configure_upgrades.tmpl.in
@@ -1,3 +1,8 @@
 append usr/share/anaconda/interactive-defaults.ks "%post --erroronfail"
 append usr/share/anaconda/interactive-defaults.ks "sed -i 's/container-image-reference=.*/container-image-reference=ostree-image-signed:docker:\/\/@IMAGE_REPO_ESCAPED@\/@IMAGE_NAME@:@VERSION@/' /ostree/deploy/default/deploy/*.origin"
-append usr/share/anaconda/interactive-defaults.ks "%end"
\ No newline at end of file
+append usr/share/anaconda/interactive-defaults.ks "%end"
+
+append usr/share/anaconda/ "%post --erroronfail"
+append usr/share/anaconda/interactive-defaults.ks "sed -i 's/container-image-reference=.*/container-image-reference=ostree-image-signed:docker:\/\/@IMAGE_REPO_ESCAPED@\/@IMAGE_NAME@:@VERSION@/' /ostree/deploy/default/deploy/*.origin"
+append usr/share/anaconda/post-scripts/container-image-reference.ks "sed -i 's/container-image-reference=.*/container-image-reference=ostree-image-signed:docker:\/\/@IMAGE_REPO_ESCAPED@\/@IMAGE_NAME@:@VERSION@/' /ostree/deploy/default/deploy/*.origin"
+append usr/share/anaconda/interactive-defaults.ks "%end"