Fix code context not showing the correct context when first toggled on.