* image download completion.
         */
        ee = mhi_get_exec_env(mhi_cntrl);
+       if (ee == MHI_EE_MAX)
+               goto error_exit_rddm;
+
        if (ee != MHI_EE_RDDM) {
                dev_dbg(dev, "Trigger device into RDDM mode using SYS ERR\n");
                mhi_set_mhi_state(mhi_cntrl, MHI_STATE_SYS_ERR);
        ee = mhi_get_exec_env(mhi_cntrl);
        ret = mhi_read_reg(mhi_cntrl, base, BHIE_RXVECSTATUS_OFFS, &rx_status);
 
-       dev_err(dev, "Did not complete RDDM transfer\n");
-       dev_err(dev, "Current EE: %s\n", TO_MHI_EXEC_STR(ee));
        dev_err(dev, "RXVEC_STATUS: 0x%x\n", rx_status);
 
+error_exit_rddm:
+       dev_err(dev, "RDDM transfer failed. Current EE: %s\n",
+               TO_MHI_EXEC_STR(ee));
+
        return -EIO;
 }