Switch to Diddy, then swim straight up through the passage.
Swim above and right past Lockjaw, then when he's gone, swim down.
Swim above Lockjaw, then move right quickly.