diff --git a/data/data/agent/files/usr/local/bin/get-container-images.sh b/data/data/agent/files/usr/local/bin/get-container-images.sh
index 899ed2267e0..7dc05c9ee70 100644
--- a/data/data/agent/files/usr/local/bin/get-container-images.sh
+++ b/data/data/agent/files/usr/local/bin/get-container-images.sh
@@ -15,5 +15,5 @@ CONTROLLER_IMAGE=$(image_for agent-installer-csr-approver)
 INSTALLER_IMAGE=$(image_for agent-installer-orchestrator)
 AGENT_DOCKER_IMAGE=$(image_for agent-installer-node-agent)
 AGENT_INSTALLER_UTILS_IMAGE=$(image_for agent-installer-utils)
-INSTALLER_UI_IMAGE=localhost/agent-installer-ui:latest
+INSTALLER_UI_IMAGE=$(image_for agent-installer-ui)
 EOF
diff --git a/data/data/agent/systemd/units/agent-extract-tui.service b/data/data/agent/systemd/units/agent-extract-tui.service
index bdf464a7d42..e20f02046ea 100644
--- a/data/data/agent/systemd/units/agent-extract-tui.service
+++ b/data/data/agent/systemd/units/agent-extract-tui.service
@@ -7,10 +7,6 @@ ConditionPathExists=/etc/assisted/interactive-ui
 
 [Service]
 Type=oneshot
-# Temporarily load assisted-install-ui here. It was previously in agent-extract-tui.service in the
-# agent-install-utils repo but that service is being removed. 
-# TODO: Remove after assisted-install-ui is productized.
-ExecStartPre=/bin/bash -c "podman tag $(podman load -q -i /run/media/iso/images/agent-installer-ui/agent-installer-ui.tar | awk '{print $NF}') localhost/agent-installer-ui:latest"
 ExecStart=/usr/local/bin/agent-extract-tui.sh
 TimeoutStartSec=300s