diff --git a/configs/nitrogen8mm_som_2g_defconfig b/configs/nitrogen8mm_som_2g_defconfig
index c4940c587ff85be90553764b144ebfca95c6ac65..1a1036f27d0cf995be5ac00b70169997cc34bce5 100644
--- a/configs/nitrogen8mm_som_2g_defconfig
+++ b/configs/nitrogen8mm_som_2g_defconfig
@@ -18,7 +18,8 @@ CONFIG_DISTRO_DEFAULTS=y
 CONFIG_NR_DRAM_BANKS=1
 CONFIG_FIT=y
 CONFIG_SPL_LOAD_FIT=y
-CONFIG_SYS_EXTRA_OPTIONS="IMX_CONFIG=arch/arm/mach-imx/spl_sd.cfg,SPL_TEXT_BASE=0x7E1000,DDR_MB=2048,DEFCONFIG=\"nitrogen8mm_som_2g\""
+CONFIG_SPL_FIT_GENERATOR="arch/arm/mach-imx/mkimage_fit_atf.sh"
+CONFIG_SYS_EXTRA_OPTIONS="IMX_CONFIG=arch/arm/mach-imx/imx8m/imximage.cfg,SPL_TEXT_BASE=0x7E1000,DDR_MB=2048,DEFCONFIG=\"nitrogen8mm_som_2g\""
 CONFIG_BOOTSTAGE_STASH_SIZE=4096
 CONFIG_ARCH_MISC_INIT=y
 CONFIG_SPL_BOARD_INIT=y
diff --git a/configs/nitrogen8mm_som_4g_defconfig b/configs/nitrogen8mm_som_4g_defconfig
index 74e830f840f1303a2fb3f45311dab070460aed5b..bdff87f1f322ea71e059d3d3618e06c08e850206 100644
--- a/configs/nitrogen8mm_som_4g_defconfig
+++ b/configs/nitrogen8mm_som_4g_defconfig
@@ -18,7 +18,8 @@ CONFIG_DISTRO_DEFAULTS=y
 CONFIG_NR_DRAM_BANKS=1
 CONFIG_FIT=y
 CONFIG_SPL_LOAD_FIT=y
-CONFIG_SYS_EXTRA_OPTIONS="IMX_CONFIG=arch/arm/mach-imx/spl_sd.cfg,SPL_TEXT_BASE=0x7E1000,DDR_MB=4096,DEFCONFIG=\"nitrogen8mm_som_4g\""
+CONFIG_SPL_FIT_GENERATOR="arch/arm/mach-imx/mkimage_fit_atf.sh"
+CONFIG_SYS_EXTRA_OPTIONS="IMX_CONFIG=arch/arm/mach-imx/imx8m/imximage.cfg,SPL_TEXT_BASE=0x7E1000,DDR_MB=4096,DEFCONFIG=\"nitrogen8mm_som_4g\""
 CONFIG_BOOTSTAGE_STASH_SIZE=4096
 CONFIG_ARCH_MISC_INIT=y
 CONFIG_SPL_BOARD_INIT=y