diff --git a/xCAT-test/autotest/testcase/infiniband/rhel-infiniband-diskless-install.sh b/xCAT-test/autotest/testcase/infiniband/rhel-infiniband-diskless-install.sh
index d9eb79eac..18f61a70e 100755
--- a/xCAT-test/autotest/testcase/infiniband/rhel-infiniband-diskless-install.sh
+++ b/xCAT-test/autotest/testcase/infiniband/rhel-infiniband-diskless-install.sh
@@ -142,6 +142,13 @@ chmod 0755 "/install/custom/netboot/rh/mlnx.${LINUX_DISTRO%%.*}.${LINUX_ARCH}.po
 rm -rf "${OSIMAGE_OTHERPKGDIR}"
 mkdir -p "${OSIMAGE_OTHERPKGDIR}"
 
+mkdir -p "${OSIMAGE_OTHERPKGDIR}"/dkms
+[ -f "${DKMS_RPM}" ]
+[ "$?" -ne "0" ] && echo "File ${DKMS_RPM} not found." >&2 && exit 1
+cp "${DKMS_RPM}" "${OSIMAGE_OTHERPKGDIR}/dkms"
+
+( cd "${OSIMAGE_OTHERPKGDIR}" && createrepo . )
+
 rm -rf "${OSIMAGE_ROOTIMGDIR}"
 
 genimage "${OSIMAGE_NAME}"
@@ -170,7 +177,7 @@ sleep 5
 xdsh "${COMPUTE_NODE}" date
 [ "$?" -ne "0" ] && echo "Failed connect to compute node via SSH." >&2 && exit 1
 
-#xdsh "${COMPUTE_NODE}" 'rpm -q cuda' | grep ': cuda-'
-#[ "$?" -ne "0" ] && echo "CUDA installation checking failed" >&2 && exit 1
+xdsh "${COMPUTE_NODE}" 'rpm -qa'  | grep 'mlnx'
+[ "$?" -ne "0" ] && echo "MLNX OFED installation checking failed" >&2 && exit 1
 
 exit 0