After C.J. dives in to rescue Maroni during his stunt, it cuts back to the pier, and she is visible among the watching crowd.
C.J. leaves her float on the surface when she dives in to rescue Maroni. But in one underwater shot, she is removing the strap from her shoulder.