Baby rolling and getting arm stuck

My son has just started rolling from back to belly but he doesn’t always know what to do with his arm and one arm ends up stuck under his belly/chest. When he does this at night I worry that he’ll cut off the circulation. Should I go roll him back to his back each time until he figures out how to get his arm free? I know once he can roll to tummy it’s safe for sleep but I just worry about his arm. I watched him on the monitor struggling to try to get it out earlier and then he fell asleep with it stuck. I did go in to roll him but he woke up and then took a while to fall back asleep because he’d seen me.