Fix context ref