A UMU can be used if the unit is completely underwater, submerged, but would have trouble if it goes above waterline. IE: it's head.
I see no known reason why it shouldn't be allowed. But suspect the usage to be the jumping out of water.
I see using it to soft land into water, then sink to depth 2 to use the UMU.
TT