#define IVPU_TEST_MODE_NULL_SUBMISSION    BIT(2)
 #define IVPU_TEST_MODE_D0I3_MSG_DISABLE   BIT(4)
 #define IVPU_TEST_MODE_D0I3_MSG_ENABLE    BIT(5)
-#define IVPU_TEST_MODE_PREEMPTION_DISABLE BIT(6)
+#define IVPU_TEST_MODE_MIP_DISABLE        BIT(6)
 #define IVPU_TEST_MODE_DISABLE_TIMEOUTS   BIT(8)
 #define IVPU_TEST_MODE_TURBO             BIT(9)
 extern int ivpu_test_mode;
 
        u64 primary_size = ALIGN(vdev->fw->primary_preempt_buf_size, PAGE_SIZE);
        u64 secondary_size = ALIGN(vdev->fw->secondary_preempt_buf_size, PAGE_SIZE);
 
-       if (vdev->fw->sched_mode != VPU_SCHEDULING_MODE_HW)
+       if (vdev->fw->sched_mode != VPU_SCHEDULING_MODE_HW ||
+           ivpu_test_mode & IVPU_TEST_MODE_MIP_DISABLE)
                return 0;
 
        cmdq->primary_preempt_buf = ivpu_bo_create(vdev, &file_priv->ctx, &vdev->hw->ranges.user,
        if (unlikely(ivpu_test_mode & IVPU_TEST_MODE_NULL_SUBMISSION))
                entry->flags = VPU_JOB_FLAGS_NULL_SUBMISSION_MASK;
 
-       if (vdev->fw->sched_mode == VPU_SCHEDULING_MODE_HW &&
-           (unlikely(!(ivpu_test_mode & IVPU_TEST_MODE_PREEMPTION_DISABLE)))) {
+       if (vdev->fw->sched_mode == VPU_SCHEDULING_MODE_HW) {
                if (cmdq->primary_preempt_buf) {
                        entry->primary_preempt_buf_addr = cmdq->primary_preempt_buf->vpu_addr;
                        entry->primary_preempt_buf_size = ivpu_bo_size(cmdq->primary_preempt_buf);