How I would approach it.

1/ Focus on output / outcome. If the proof backs you up, then keep doing it.

2/ Schedule it in your calendar. If it’s in the calendar, then it’s work. But you’re doing it in the way that supports your creativity better. See point 1.