Fixed UIContext usage bug