correct bugs on ui updates