My 2 month old baby can only sleep when being held

So my baby stayed at the hospital for 27 days due to blood infection and I was there with him throughout (because I was breastfeeding him). He used to sleep on the hospital’s crib (most of the time getting startled with the noise from the ward) but other than that, he can sleep on his crib. Now when we got discharged and already at home, I tried putting him to sleep and settle him down on his bassinet but he wakes up the moment I lay him down. Now at 2 months, he refuses to sleep anywhere but on me. He wants to be held when sleeping and I fear that something might happen to him if I accidentally doze off. It’s been affecting my sleep and I couldn’t get anything done because I can’t put him down when sleeping. Any tips?