So would you pass `context` into the `before` and `after` blocks as well?