diff --git a/meta-oniro-core/classes/oniro-image.bbclass b/meta-oniro-core/classes/oniro-image.bbclass
index cbf3e70d75c32c124d839dee07d998394fd74dfe..a21a06b778d5e01a54c41489dd772608ad113c52 100644
--- a/meta-oniro-core/classes/oniro-image.bbclass
+++ b/meta-oniro-core/classes/oniro-image.bbclass
@@ -48,7 +48,8 @@ WKS_FILE_DEPENDS:qemux86 += "ovmf"
 WKS_FILE:qemu-generic-arm64 = "x-qemu-efi-disk.wks.in"
 IMAGE_FSTYPES:qemu-generic-arm64 += "wic wic.qcow2"
 
-WKS_FILE:seco-imx8mm-c61 ?= "x-imx-uboot-bootpart.wks.in"
+WKS_FILE:seco-imx8mm-c61-2gb ?= "x-imx-uboot-bootpart.wks.in"
+WKS_FILE:seco-imx8mm-c61-4gb ?= "x-imx-uboot-bootpart.wks.in"
 
 #
 # Deploy boot partition artifacts to the root partition. This is part of the