An assembly for a drug delivery device is provided, wherein the assembly comprises an actuator which is configured to perform a setting movement in order to set a dose and which is configured to perform a dispense movement in order to dispense a dose and a piston rod which is configured to be moved in a distal direction in order to dispense a dose, and a snap feature which is configured to snap into engagement with the piston rod. The snap feature may be configured to inhibit a dispense movement of the actuator and/or a setting movement of the actuator when a maximum amount of medication has been delivered.