We went through the same exact thing. I just kept flipping DD over onto her back until she fell asleep. Sometimes I would prop her up on a pillow for naps because she couldn't flip over like that. When I did that I kept a very close eye on her the whole time. It didn't take very long for DD to learn how to flip over from stomach to back and the problem went away.