Yeah, once it gives you a couple of written messages that your lung capacity's upgraded, you'll be good to go and the cutscene will be completely different. Lung capacity and swimming are the same thing in San An - You'll need to stay underwater for quite a while in this mission.