is there any chance we could move this to `get_context` ? We use underscores consistently :)