I need advise or tips or suggestions! Anything at this point. I'm going to try to explain this the best I can lol

So my 6 month old son has this obsession with being on his belly. Loves to army crawl around. He can flip both ways but doesn't ever really flip on to his back from tummy because he just wants to be on his tummy.

Problem is at bed time he will continuously roll on to his belly. Sometimes he screams instantly and sometimes he falls asleep but wakes up 20mins to 1.5 hrs later screaming. It's like he wants to be on his belly but hates it at the same time. Sometimes he will fall sleep on his back but then in the middle of the night roll on his belly and start screaming again. I will go in a flip him over but then he just tries to roll on his belly again. At the beginning I would roll him back over on his back until he fell asleep that way which sometimes took me 15 times. But now he's getting pissed when I roll him onto his back but he's pissed on his belly to.... Onces he screaming he takes 20mins to an hour to get back down. He's not hungry, clean diaper...he's just pissed off lol

BUT if he stays on his back he will sleep all night.... What do I doooooooo?!?

Picture of the cutie just because lol hope this post made sense lol