Tweak context ui a bit