Night shift pregnant mamas

I’m 6 weeks and a night shift nurse. I have the hardest time staying asleep after a night shift and a very hard time staying up at night. I’ve been doing nights for 7 years and I’m very use to it but pregnancy has got me completely thrown off! Any advice?