Duplicate the action set, but add "Note: Off" condition. For example, if you have:
1) Track current Mute
2) Toggle value
add
3) Note:Off - Track current Mute
4) Note:Off - Toggle value
Plug-in detect "Note off" and normally does nothing on it (most buttons are momentary, so that is "awaited"). Note, that in case the signal is "CC", or "Note: On Value 0", you can just switch "Hardware type" to let say "Slider" to get the same effect (rotaries and sliders use Value 0 as normal). But in case you really getting "Note Off" MIDI message, described addition is the only way.