Swaddling baby

My LO has just started rolling to her tummy on her own and as I understand this is a cue that I should stop swaddling her to sleep and use a sleeping bag instead , but her startle reflex is still very strong. So how should I go about this ?