Fix context on callbacks