diff --git a/pulser-core/pulser/sequence/_schedule.py b/pulser-core/pulser/sequence/_schedule.py
index f2cad62ed..816c5c912 100644
--- a/pulser-core/pulser/sequence/_schedule.py
+++ b/pulser-core/pulser/sequence/_schedule.py
@@ -24,6 +24,7 @@
 from pulser.channels.base_channel import Channel
 from pulser.pulse import Pulse
 from pulser.register.base_register import QubitId
+from pulser.register.weight_maps import DetuningMap
 from pulser.sampler.samples import ChannelSamples, _PulseTargetSlot
 from pulser.waveforms import ConstantWaveform
 
@@ -247,6 +248,11 @@ def __iter__(self) -> Iterator[_TimeSlot]:
             yield slot
 
 
+@dataclass
+class _DMMSchedule(_ChannelSchedule):
+    detuning_map: DetuningMap
+
+
 class _Schedule(Dict[str, _ChannelSchedule]):
     def __init__(self, max_duration: int | None = None):
         self.max_duration = max_duration