I find myself getting carried away and working late sometimes - How can I have better WLB?

• Quickest tactic: Set a recurring daily alarm to tell yourself to stop working. From there, it's up to you to respect the alarm.
• Having clean, healthy work/life boundaries is another thing that manifests as a habit. The hardest part is the beginning: Get into the groove of respecting that groove/time cut-off 5-10 times and actually stopping work, and from there, your brain will start automatically doing it.
• A more extreme option is to insert hard commitments to stop yourself from working. For example, you want to stop working at 7PM, so you set up dinner with a friend around that time.
• Make sure that the technology is cleanly split as well. Don't look at your work laptop after working hours. Mute work notifications. Pretend your work devices don't exist after working hours (unless you are oncall).

Related resources:

• "How can I have less burnout and better work-life balance?"
• "How does work-life balance shift across levels?"
• "How can I pursue growth while maintaining work-life balance?"

In my experience, what works best for me is to block off times on work my calendar dedicated for

• workout
• cooking
• winding down It acts both as a reminder to myself and also a note to co-workers.

I also encourage you to take a step back what is the problem you are trying to solve here. WLB is a tricky word because it's you that need to find the balance. 9 - 5 is not necessarily what you should aim for based on your circumstance.

Sounds like you are in a flow state working on something you are passionate about. That's a good thing. If it's impacting your sleep because of you cannot turn off your mind soon enough, work on that.