Theo wasn't expecting the insistent pushing to ease back and when he broke the kiss he echoed that confusion in sound, tilting his head to watch Greg pull back. He wobbled a little to keep his perch on Greg's lap when his boyfriend leaned back, reaching back when Greg then reached forward, expecting another cuddle.
That wasn't what happened and he tensed a little in anticipation of what he didn't know, but his expectations of more skin to skin contact were proved wrong as Greg disappeared... Theo panicked for a second before he saw Greg resurface and then swim away.
He watched Greg go but shivered even in the water, sidling across and huddling up against the side of the steps, arms crossed and his body bent down over his knees, trembling cold as he watched Greg and tried to make sense of the distance.