You should not actually stay in bed for very long awake, because your brain is this remarkably associative device, and it quickly learns that the bed is about being awake. So you should go to another room - a room that's dim. Just read a book - no screens, no phones - and, only when you're sleepy, return to the bed.