think out of the box, just before water impact, perhaps a parachute deploys, and some transhab bags inflate to keep most of the booster from being submerged..

of course the booster could be built with salt water resistant materials.