Small improvement on the machineState button