diff --git a/images/ubuntu/scripts/build/configure-apt-sources.sh b/images/ubuntu/scripts/build/configure-apt-sources.sh
index b5afda474..1e6c98220 100644
--- a/images/ubuntu/scripts/build/configure-apt-sources.sh
+++ b/images/ubuntu/scripts/build/configure-apt-sources.sh
@@ -4,13 +4,22 @@
 ##  Desc:  Configure apt sources with failover from Azure to Ubuntu archives.
 ################################################################################
 
+source $HELPER_SCRIPTS/os.sh
+
 touch /etc/apt/apt-mirrors.txt
 
 printf "http://azure.archive.ubuntu.com/ubuntu/\tpriority:1\n" | tee -a /etc/apt/apt-mirrors.txt
 printf "http://archive.ubuntu.com/ubuntu/\tpriority:2\n" | tee -a /etc/apt/apt-mirrors.txt
 printf "http://security.ubuntu.com/ubuntu/\tpriority:3\n" | tee -a /etc/apt/apt-mirrors.txt
 
-sed -i 's/http:\/\/azure.archive.ubuntu.com\/ubuntu\//mirror+file:\/etc\/apt\/apt-mirrors.txt/' /etc/apt/sources.list
+if is_ubuntu24; then
+    sed -i 's/http:\/\/azure.archive.ubuntu.com\/ubuntu\//mirror+file:\/etc\/apt\/apt-mirrors.txt/' /etc/apt/sources.list.d/ubuntu.sources
 
-# Apt changes to survive Cloud Init
-cp -f /etc/apt/sources.list /etc/cloud/templates/sources.list.ubuntu.tmpl
+    # Apt changes to survive Cloud Init
+    cp -f /etc/apt/sources.list.d/ubuntu.sources  /etc/cloud/templates/sources.list.ubuntu.deb822.tmpl
+else
+    sed -i 's/http:\/\/azure.archive.ubuntu.com\/ubuntu\//mirror+file:\/etc\/apt\/apt-mirrors.txt/' /etc/apt/sources.list
+
+    # Apt changes to survive Cloud Init
+    cp -f /etc/apt/sources.list /etc/cloud/templates/sources.list.ubuntu.tmpl
+fi