UI task fix