Dear Davy:
Regarding your bottom line question, my answer: of the two possibilities, do accept thinking about work. But to that I add: introduce to your after work time thinking about something new, something meaningful to you that is not work related.
You think about work because it is meaningful to you, you wrote that you love what you do. You need something else that you will love to do, something that intrigues you, that draws you.
(There may be- or not- something at home that repels you, maybe a feeling of loneliness or if in a relationship, maybe conflicts which lead you to the distraction of thoughts of work?)
anita