Use context already passed in `onMessage`