Rolling over and sleeping.

hey guys, it's been awhile since I've posted on here and I need some advice. my four month old has been sleeping in her crib, which was great and she has made it back in my bed because now she actively rolls over to her stomach while shes asleep and sometimes it wakes her up and she starts to cry. she doesn't roll from front to back yet. I deff don't want to restrain her but I also don't want to keep getting up in the middle of the night so she has been in my bed any suggestions?