I find myself in that situation often enough. Whenever I do, I pretty much say to myself, "Screw this, I ned the sleep" and then try to catch myself whenever I roll over or adjust the pillow.

One thing, how long are you staying awake before relaxing again? Your problem could be a combination of performance anxiety and letting yourself awaken too much.

Welcome back!