diff --git a/configs/nitrogen8mm_2g_defconfig b/configs/nitrogen8mm_2g_defconfig
index 452a224effd9a818772d768c47ef0cab317c08e3..ddccc790e569ce7771535565861ff8fab0722b1b 100644
--- a/configs/nitrogen8mm_2g_defconfig
+++ b/configs/nitrogen8mm_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_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_2g\""
 CONFIG_BOOTSTAGE_STASH_SIZE=4096
 CONFIG_ARCH_MISC_INIT=y
 CONFIG_SPL_BOARD_INIT=y
diff --git a/configs/nitrogen8mm_4g_defconfig b/configs/nitrogen8mm_4g_defconfig
index 8f1adf3791eae11c06a8bcc1f3202400fa26d632..148f32285d8a6d430e00399194a99dd1d73c18dd 100644
--- a/configs/nitrogen8mm_4g_defconfig
+++ b/configs/nitrogen8mm_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_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_4g\""
 CONFIG_BOOTSTAGE_STASH_SIZE=4096
 CONFIG_ARCH_MISC_INIT=y
 CONFIG_SPL_BOARD_INIT=y