Correction in UI controller