* - IMR boot: wait for ROM firmware entered (firmware booted up from IMR)
         */
        if (imr_boot)
-               target_status = HDA_DSP_ROM_FW_ENTERED;
+               target_status = FSR_STATE_FW_ENTERED;
        else
-               target_status = HDA_DSP_ROM_INIT;
+               target_status = FSR_STATE_INIT_DONE;
 
        ret = snd_sof_dsp_read_poll_timeout(sdev, HDA_DSP_BAR,
                                        chip->rom_status_reg, status,
-                                       ((status & HDA_DSP_ROM_STS_MASK)
-                                               == target_status),
+                                       (FSR_TO_STATE_CODE(status) == target_status),
                                        HDA_DSP_REG_POLL_INTERVAL_US,
                                        chip->rom_init_timeout *
                                        USEC_PER_MSEC);
 
        status = snd_sof_dsp_read_poll_timeout(sdev, HDA_DSP_BAR,
                                        chip->rom_status_reg, reg,
-                                       ((reg & HDA_DSP_ROM_STS_MASK)
-                                               == HDA_DSP_ROM_FW_ENTERED),
+                                       (FSR_TO_STATE_CODE(reg) == FSR_STATE_FW_ENTERED),
                                        HDA_DSP_REG_POLL_INTERVAL_US,
                                        HDA_DSP_BASEFW_TIMEOUT_US);