diff --git a/arch/arm/mach-imx/mmc_env.c b/arch/arm/mach-imx/mmc_env.c
index 9c822f7..addb50f 100644
--- a/arch/arm/mach-imx/mmc_env.c
+++ b/arch/arm/mach-imx/mmc_env.c
@@ -11,6 +11,9 @@
 
 __weak int board_mmc_get_env_dev(int devno)
 {
+	if (CONFIG_IS_ENABLED(DM_MMC))
+		return devno;
+
 	return CONFIG_SYS_MMC_ENV_DEV;
 }