If I could write a scene in Ne Zha 3, I'd reference this:
For context, this is a retelling of Ne Zha's story, where instead of killing Ao Bing, he chooses to spare him because he's a cute little dragon. Ne Zha brings Ao Bing home as a friend, and places him in a goldfish pond to recover from the injuries he'd sustained during their fight.
The way I'd reference this story would go like this:
The duo fight an enemy who's capable of manipulating the Qiankun hoop. They make it so the hoop forces Ne Zha to return to his child form and also shuts off all his power. This is an underwater battle; Ne Zha was using his fire powers to generate oxygen. Now he's drowning.
Ao Bing quickly transforms into his dragon form and swallows Ne Zha to protect him. As Ao Bing hastily swims upwards, the enemy sends a barrage of attacks, and Ao Bing is unable to dodge the hardest hit.
Upon reaching the shore, Ao Bing coughs up Ne Zha, who can finally say the necessary incantation to undo the hoop and go back to his teenage form. Ao Bing also returns to his human form, badly injured.
The enemy bursts out of the surface. Ne Zha has all six of his arms ready—one pair to wield his spear, another to manipulate his silk, and the last to carry Ao Bing. He jumps onto his fire wheels with intention to get Ao Bing to safety while fending off the enemy. Of course, this form of multitasking is easier said than done, and the enemy manages to close the distance. That's when it's revealed Ao Bing is still conscious, and he touches the enemy, encasing them in ice so they fall straight to the ground.
At last, the duo reach a pond filled with goldfish. Ne Zha lowers Ao Bing into it, as dragons heal faster when submerged in water (source: I made it up).
Ne Zha isn't able to get the hoop off and is stuck as a child who's too weak to carry either a large dragon or a human teen, so he pleads for Ao Bing to take on a smaller form. Fortunately, Ao Bing is able to make himself into the size of a snake. Ne Zha quickly gathers him in his arms and runs to safety.