Ah, I missed this in my refactor.

Originally I planned to dependency inject it, then I converted it to extract it from `context`. I'll fix it in a sec.