4 month old tummy sleeping anxiety

My 4 month old loves to sleep on her tummy or her side. I always put her to sleep on her back but she will eventually roll herself to her side or tummy to sleep. When awake, she can easily roll herself from her back to tummy one way, the other way from back to tummy she’s still working on, and from tummy to back she rarely does and is definitely still learning to do (she’s only done tummy to back a few times). My pediatrician said that if we see her on her tummy or side to gently roll her back on her back. We’ve been doing this but she will almost immediately go back on her tummy when we roll her back and it’s just a constant struggle doing this so many times throughout the night and worry on my part to let her sleep on her tummy or side when she puts herself in that position all the time. Sometimes she’ll wake up when we roll her back on her back. Would love to hear any advice or feedback! Thanks!