Clarify type of 2nd argument `context`