Swaddle transitioning

FTM how do you transition if baby still startles himself awake?

He’s 4 months next week.