fixed minor ui issue