1 files changed, 2 insertions, 2 deletions

diff --git a/meta/classes/boot-directdisk.bbclass b/meta/classes/boot-directdisk.bbclass
index 44f738b02e..c5765233ac 100644
--- a/meta/classes/boot-directdisk.bbclass
+++ b/meta/classes/boot-directdisk.bbclass
@@ -63,7 +63,7 @@ DISK_SIGNATURE ?= "${DISK_SIGNATURE_GENERATED}"
 SYSLINUX_ROOT ?= "root=/dev/sda2"
 SYSLINUX_TIMEOUT ?= "10"
 
-IS_VMDK = '${@bb.utils.contains("IMAGE_FSTYPES", "vmdk", "true", "false", d)}'
+IS_VM = '${@bb.utils.contains_any("IMAGE_FSTYPES", ["vmdk" ,"vdi"], "true", "false", d)}'
 
 boot_direct_populate() {
         dest=$1
@@ -101,7 +101,7 @@ build_boot_dd() {
                 efi_hddimg_populate $HDDDIR
         fi
 
-        if [ "${IS_VMDK}" = "true" ]; then
+        if [ "${IS_VM}" = "true" ]; then
                 if [ "x${AUTO_SYSLINUXMENU}" = "x1" ] ; then
                         install -m 0644 ${STAGING_DIR}/${MACHINE}/usr/share/syslinux/vesamenu.c32 $HDDDIR/${SYSLINUXDIR}/
                         if [ "x${SYSLINUX_SPLASH}" != "x" ] ; then