diff --git a/meta-digi-arm/recipes-bsp/u-boot/u-boot-dey/ccimx6ul/install_linux_fw_uuu.sh b/meta-digi-arm/recipes-bsp/u-boot/u-boot-dey/ccimx6ul/install_linux_fw_uuu.sh
index c6971b7b0..8c48714e0 100755
--- a/meta-digi-arm/recipes-bsp/u-boot/u-boot-dey/ccimx6ul/install_linux_fw_uuu.sh
+++ b/meta-digi-arm/recipes-bsp/u-boot/u-boot-dey/ccimx6ul/install_linux_fw_uuu.sh
@@ -87,7 +87,7 @@ uuu fb: ucmd setenv stdout serial,fastboot
 if [ -z "${INSTALL_UBOOT_FILENAME}" ]; then
 	module_variant=$(getenv "module_variant")
 	# Determine U-Boot file to program basing on SOM's variant
-	if [ -n "$module_variant" ]; then
+	if [ -n "$module_variant" || "$module_variant" = "0x00" ]; then
 		if [ "$module_variant" = "0x08" ] || \
 		   [ "$module_variant" = "0x09" ]; then
 			INSTALL_UBOOT_FILENAME="u-boot-##MACHINE##512MB.imx"
diff --git a/meta-digi-arm/recipes-bsp/u-boot/u-boot-dey/ccimx8x/install_linux_fw_uuu.sh b/meta-digi-arm/recipes-bsp/u-boot/u-boot-dey/ccimx8x/install_linux_fw_uuu.sh
index 88eed85b3..a51502e1b 100644
--- a/meta-digi-arm/recipes-bsp/u-boot/u-boot-dey/ccimx8x/install_linux_fw_uuu.sh
+++ b/meta-digi-arm/recipes-bsp/u-boot/u-boot-dey/ccimx8x/install_linux_fw_uuu.sh
@@ -101,16 +101,16 @@ if [ -z ${INSTALL_UBOOT_FILENAME} ]; then
 	if [ -z "${module_ram}" ]; then
 		module_variant=$(getenv "module_variant")
 		# Determine U-Boot file to program basing on SOM's variant
-		if [ -n "$module_variant" ]; then
-			if [ "$module_variant" == "0x01" ] || \
-			   [ "$module_variant" == "0x04" ] || \
-			   [ "$module_variant" == "0x05" ]; then
+		if [ -n "$module_variant" || "$module_variant" = "0x00" ]; then
+			if [ "$module_variant" = "0x01" ] || \
+			   [ "$module_variant" = "0x04" ] || \
+			   [ "$module_variant" = "0x05" ]; then
 				module_ram="1GB"
-			elif [ "$module_variant" == "0x02" ] || \
-			     [ "$module_variant" == "0x03" ]; then
-				module_ram="2GB"
-			else
+			elif [ "$module_variant" = "0x06" ] || \
+			     [ "$module_variant" = "0x09" ]; then
 				module_ram="512MB"
+			else
+				module_ram="2GB"
 			fi
 			INSTALL_UBOOT_FILENAME="imx-boot-##MACHINE##-${soc_rev}-${module_ram}_${bus_width}.bin"
 		fi