Once baby transitions out of the swaddle but is not ready to sleep without any support a sleepsuit or weighted sleep sack could be the answer.
MERLIN'SMagic Sleepsuit
NESTED BEANWeighted Sleepsuit
Sleep Suits | Sacks
