Fix context reference