fixing for UI

fixing for UI