Baby rolling over in sleep and waking up

Nicole • Momma 🐻

My son has been able to roll both ways for a while, but just started rolling back to belly in his sleep and wakes up SCREAMING. He can roll belly to back, but for some reason forgets (maybe because he is scared?) how to in his crib. What can I do?