4 month old still wants to swaddle

Is that okay? When should I stop. I try to put him on his back while his already asleep ( not swaddled) but he’ll only sleep like that for 5 minuets.. he sleeps on his stomach during the day