One simple trick I used when in the same situation, was to train myself to block all work-related thoughts. - "Stop! Does it help my thinking of work? No. Then stop." In the beginning you have to learn that you are actually thinking of work right now, so to adopt a pattern of blocking. Later, it gets easier.