When do I stop swaddling my baby?

i stopped swaddling at 3months old. As mine is a premature baby, he gets frighten easily