diff --git a/runtime/command_stream/command_stream_receiver_hw_base.inl b/runtime/command_stream/command_stream_receiver_hw_base.inl
index bc48ba9b2f..bb396f69c5 100644
--- a/runtime/command_stream/command_stream_receiver_hw_base.inl
+++ b/runtime/command_stream/command_stream_receiver_hw_base.inl
@@ -259,7 +259,6 @@ CompletionStamp CommandStreamReceiverHw<GfxFamily>::flushTask(
     }
 
     initPageTableManagerRegisters(commandStreamCSR);
-    programPreemption(commandStreamCSR, dispatchFlags);
     programComputeMode(commandStreamCSR, dispatchFlags);
     programL3(commandStreamCSR, dispatchFlags, newL3Config);
     programPipelineSelect(commandStreamCSR, dispatchFlags.pipelineSelectArgs);
@@ -275,6 +274,8 @@ CompletionStamp CommandStreamReceiverHw<GfxFamily>::flushTask(
 
     programVFEState(commandStreamCSR, dispatchFlags, device.getDeviceInfo().maxFrontEndThreads);
 
+    programPreemption(commandStreamCSR, dispatchFlags);
+
     bool dshDirty = dshState.updateAndCheck(&dsh);
     bool iohDirty = iohState.updateAndCheck(&ioh);
     bool sshDirty = sshState.updateAndCheck(&ssh);