diff --git a/configs/airootfs/root/.automated_script.sh b/configs/airootfs/root/.automated_script.sh
index 5df28fc..9878419 100644
--- a/configs/airootfs/root/.automated_script.sh
+++ b/configs/airootfs/root/.automated_script.sh
@@ -124,9 +124,8 @@ if [[ $(tty) == "/dev/tty1" ]]; then
   fi
 
   source "$OMARCHY_INSTALL/preflight/set-size-vars.sh"
-  source "$OMARCHY_INSTALL/helpers/ansi-codes.sh"
   source "$OMARCHY_INSTALL/helpers/logo.sh"
-  source "$OMARCHY_INSTALL/preflight/set-gum-styling.sh"
+  source "$OMARCHY_INSTALL/preflight/gum.sh"
   source "$OMARCHY_INSTALL/helpers/tail-log-output.sh"
   source "$OMARCHY_INSTALL/helpers/trap-errors.sh"