You can definetely put it on while in the water. Trick would be to remain calm and not drop the plug while putting it on under water.

Sent from my SM-G960U1 using Tapatalk