How do you get back to sleep after waking up in the middle of the night?

So first let me say that before I go to bed I drink some water, have a couple puffs off my vape and go pee so I dont wake up to go pee every hour. When I wake up from nightmares I go pee, have a puff from my vape to help me calm down, drink a couple sips of water and go back to bed. I tend to repeat my right before getting into bed routine to trick my mind into getting tired again.